michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

fixed naming

Browse Source
This commit is contained in:
Liyu Xiao 2025-06-16 10:17:17 -04:00
parent 95d56b90a2
commit 586acd123c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
onnx_scripts/convert_to_onnx.py
View File

@ -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))