diff options
| -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!") |