This commit is contained in:
parent
b4d0ffa58f
commit
abb65e3514
|
@ -80,6 +80,6 @@ if __name__ == "__main__":
|
||||||
ckpt_path=os.path.join("checkpoint_files", model_checkpoint), fp16=False
|
ckpt_path=os.path.join("checkpoint_files", model_checkpoint), fp16=False
|
||||||
)
|
)
|
||||||
|
|
||||||
output_file = "convert_to_onnx.py" + ".onnx"
|
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_scripts", output_file))
|
||||||
|
|
Loading…
Reference in New Issue
Block a user