From 95d56b90a28db96cc5a879f7c6085fd5115df736 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Mon, 16 Jun 2025 10:13:55 -0400 Subject: [PATCH] debugging --- onnx_scripts/convert_to_onnx.py | 1 + 1 file changed, 1 insertion(+) 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(