diff options
Diffstat (limited to 'training')
| -rwxr-xr-x | training/export_cnn_v2_shader.py | 2 | ||||
| -rwxr-xr-x | training/export_cnn_v2_weights.py | 2 |
2 files changed, 2 insertions, 2 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}") diff --git a/training/export_cnn_v2_weights.py b/training/export_cnn_v2_weights.py index 723f572..d8c7c10 100755 --- a/training/export_cnn_v2_weights.py +++ b/training/export_cnn_v2_weights.py @@ -248,7 +248,7 @@ fn main(@builtin(global_invocation_id) id: vec3<u32>) { } """ - output_path = Path(output_dir) / "cnn_v2_compute.wgsl" + output_path = Path(output_dir) / "cnn_v2" / "cnn_v2_compute.wgsl" output_path.write_text(shader_code) print(f" → {output_path}") |