summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
Diffstat (limited to 'training')
-rwxr-xr-xtraining/export_cnn_v2_weights.py3
1 files changed, 2 insertions, 1 deletions
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!")