Skip to content

Commit d788c1a

Browse files
authored
Merge pull request #11 from writer-in-fancy-pants/patch-1
Fixing ONNX error
2 parents 83a2217 + c79be86 commit d788c1a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/convert-pth-to-onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@
4848
torch.onnx.export(
4949
core_model,
5050
dummy_input,
51-
onnx_file_path,
51+
str(onnx_file_path),
5252
export_params=True,
5353
opset_version=17,
5454
do_constant_folding=True,

0 commit comments

Comments
 (0)