diff options
| author | skal <pascal.massimino@gmail.com> | 2026-02-13 08:24:36 +0100 |
|---|---|---|
| committer | skal <pascal.massimino@gmail.com> | 2026-02-13 08:24:36 +0100 |
| commit | 064ebb52847949d62f81873f5e44f12bca4d2e60 (patch) | |
| tree | 217fb2a45b1d4a76d6ed3d0e062085fe39cc0f5e /training/export_cnn_v2_weights.py | |
| parent | eb15703a3f87e4eadc8839b06de12b9c6ec54023 (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-x | training/export_cnn_v2_weights.py | 2 |
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') |
