debugging

2025-06-16 10:13:55 -04:00 · 2025-06-16 10:13:55 -04:00 · 95d56b90a2
commit 95d56b90a2
parent 32642468c9
1 changed files with 1 additions and 0 deletions
--- a/onnx_scripts/convert_to_onnx.py
+++ b/onnx_scripts/convert_to_onnx.py
@ -53,6 +53,7 @@ def convert_to_onnx(ckpt_path, fp16=False):
        )
    else:
        output_path = os.path.join("onnx_scripts", f"{base}.onnx")
+        print(output_path)
        sample_input = torch.rand(batch_size, in_channels, slice_length)

    torch.onnx.export(