summaryrefslogtreecommitdiff
path: root/training/export_cnn_v2_shader.py
diff options
context:
space:
mode:
Diffstat (limited to 'training/export_cnn_v2_shader.py')
-rwxr-xr-xtraining/export_cnn_v2_shader.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/export_cnn_v2_shader.py b/training/export_cnn_v2_shader.py
index 3c53ce2..add28d2 100755
--- a/training/export_cnn_v2_shader.py
+++ b/training/export_cnn_v2_shader.py
@@ -144,7 +144,7 @@ fn main(@builtin(global_invocation_id) id: vec3<u32>) {{
}}
"""
- output_path = Path(output_dir) / f"cnn_v2_layer_{layer_idx}.wgsl"
+ output_path = Path(output_dir) / "cnn_v2" / f"cnn_v2_layer_{layer_idx}.wgsl"
output_path.write_text(shader_code)
print(f" → {output_path}")