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

debugging

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

1
onnx_scripts/convert_to_onnx.py
View File

@ -53,6 +53,7 @@ def convert_to_onnx(ckpt_path, fp16=False):
) )
else: else:
output_path = os.path.join("onnx_scripts", f"{base}.onnx") output_path = os.path.join("onnx_scripts", f"{base}.onnx")
print(output_path)
sample_input = torch.rand(batch_size, in_channels, slice_length) sample_input = torch.rand(batch_size, in_channels, slice_length)
torch.onnx.export( torch.onnx.export(