summaryrefslogtreecommitdiff
path: root/training/export_cnn_v2_weights.py
diff options
context:
space:
mode:
authorskal <pascal.massimino@gmail.com>2026-02-13 08:24:36 +0100
committerskal <pascal.massimino@gmail.com>2026-02-13 08:24:36 +0100
commit064ebb52847949d62f81873f5e44f12bca4d2e60 (patch)
tree217fb2a45b1d4a76d6ed3d0e062085fe39cc0f5e /training/export_cnn_v2_weights.py
parenteb15703a3f87e4eadc8839b06de12b9c6ec54023 (diff)
Add weights/ subdirectory to workspaces for CNN training outputs
Each workspace now has a weights/ directory to store binary weight files from CNN training (e.g., cnn_v2_weights.bin). Changes: - Created workspaces/{main,test}/weights/ - Moved cnn_v2_weights.bin → workspaces/main/weights/ - Updated assets.txt reference - Updated training scripts and export tool paths handoff(Claude): Workspace weights/ directories added
Diffstat (limited to 'training/export_cnn_v2_weights.py')
-rwxr-xr-xtraining/export_cnn_v2_weights.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/export_cnn_v2_weights.py b/training/export_cnn_v2_weights.py
index d8c7c10..8a2fcdc 100755
--- a/training/export_cnn_v2_weights.py
+++ b/training/export_cnn_v2_weights.py
@@ -256,7 +256,7 @@ fn main(@builtin(global_invocation_id) id: vec3<u32>) {
def main():
parser = argparse.ArgumentParser(description='Export CNN v2 weights to binary format')
parser.add_argument('checkpoint', type=str, help='Path to checkpoint .pth file')
- parser.add_argument('--output-weights', type=str, default='workspaces/main/cnn_v2_weights.bin',
+ parser.add_argument('--output-weights', type=str, default='workspaces/main/weights/cnn_v2_weights.bin',
help='Output binary weights file')
parser.add_argument('--output-shader', type=str, default='workspaces/main/shaders',
help='Output directory for shader template')