debugging
This commit is contained in:
parent
32642468c9
commit
95d56b90a2
|
@ -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(
|
||||
|
|
Loading…
Reference in New Issue
Block a user