From 586acd123c0b3b74f1451267f8bf7e32d1385899 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Mon, 16 Jun 2025 10:17:17 -0400 Subject: [PATCH] fixed naming --- onnx_scripts/convert_to_onnx.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/onnx_scripts/convert_to_onnx.py b/onnx_scripts/convert_to_onnx.py index b9b17dc..7ce3696 100644 --- a/onnx_scripts/convert_to_onnx.py +++ b/onnx_scripts/convert_to_onnx.py @@ -47,12 +47,12 @@ def convert_to_onnx(ckpt_path, fp16=False): # Generate random sample data base, ext = os.path.splitext(os.path.basename(ckpt_path)) if fp16: - output_path = os.path.join("onnx_scripts", f"{base}.onnx") + output_path = os.path.join("onnx_files", f"{base}.onnx") sample_input = torch.from_numpy( np.random.rand(batch_size, in_channels, slice_length).astype(np.float16) ) else: - output_path = os.path.join("onnx_scripts", f"{base}.onnx") + output_path = os.path.join("onnx_files", f"{base}.onnx") print(output_path) sample_input = torch.rand(batch_size, in_channels, slice_length) @@ -83,4 +83,4 @@ if __name__ == "__main__": output_file = "convert_to_onnx" + ".onnx" - print("Conversion complete stored at: ", os.path.join("onnx_scripts", output_file)) + print("Conversion complete stored at: ", os.path.join("onnx_files", output_file))