diff --git a/onnx_scripts/convert_to_onnx.py b/onnx_scripts/convert_to_onnx.py index 8fb17d4..b9b17dc 100644 --- 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(