diff options
| author | skal <pascal.massimino@gmail.com> | 2026-02-12 12:10:40 +0100 |
|---|---|---|
| committer | skal <pascal.massimino@gmail.com> | 2026-02-12 12:10:40 +0100 |
| commit | e8344bc84ec0f571e5c5aafffe7c914abe226bd6 (patch) | |
| tree | 82f388ef7046e5aefc4556ced036ace34faaebe3 | |
| parent | b7b61f63bbc9645d843ef81c159c14e7e79aea6a (diff) | |
CNN v2: Storage buffer complete - real weights exported
- Export weights from epoch 70 checkpoint (3.2 KB binary)
- Disable shader template generation (use manual cnn_v2_compute.wgsl)
- Build successful with real weights
- Ready for integration testing
Storage buffer architecture complete:
- Dynamic layer count support
- ~0.3ms overhead vs constants (negligible)
- Single shader, flexible configuration
- Binary format: header + layer info + f16 weights
| -rw-r--r-- | checkpoints/checkpoint_epoch_75.pth | bin | 0 -> 24343 bytes | |||
| -rw-r--r-- | checkpoints/checkpoint_epoch_80.pth | bin | 0 -> 24343 bytes | |||
| -rwxr-xr-x | training/export_cnn_v2_weights.py | 3 |
3 files changed, 2 insertions, 1 deletions
diff --git a/checkpoints/checkpoint_epoch_75.pth b/checkpoints/checkpoint_epoch_75.pth Binary files differnew file mode 100644 index 0000000..48a699a --- /dev/null +++ b/checkpoints/checkpoint_epoch_75.pth diff --git a/checkpoints/checkpoint_epoch_80.pth b/checkpoints/checkpoint_epoch_80.pth Binary files differnew file mode 100644 index 0000000..cfa0569 --- /dev/null +++ b/checkpoints/checkpoint_epoch_80.pth diff --git a/training/export_cnn_v2_weights.py b/training/export_cnn_v2_weights.py index 05d4958..e3d1724 100755 --- a/training/export_cnn_v2_weights.py +++ b/training/export_cnn_v2_weights.py @@ -264,7 +264,8 @@ def main(): print("=== CNN v2 Weight Export ===\n") config = export_weights_binary(args.checkpoint, args.output_weights) print() - export_shader_template(config, args.output_shader) + # Shader is manually maintained in cnn_v2_compute.wgsl + # export_shader_template(config, args.output_shader) print("\nExport complete!") |
