There was an error while loading. Please reload this page.
2 parents 83a2217 + c79be86 commit d788c1aCopy full SHA for d788c1a
1 file changed
scripts/convert-pth-to-onnx.py
@@ -48,7 +48,7 @@
48
torch.onnx.export(
49
core_model,
50
dummy_input,
51
- onnx_file_path,
+ str(onnx_file_path),
52
export_params=True,
53
opset_version=17,
54
do_constant_folding=True,
0 commit comments