We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents cf8f53f + d493cd3 commit 9cd26e9Copy full SHA for 9cd26e9
pytorch2keras/converter.py
@@ -69,4 +69,7 @@ def pytorch_to_keras(
69
stream.seek(0)
70
onnx_model = onnx.load(stream)
71
72
+ k_model = onnx_to_keras(onnx_model=onnx_model, input_names=input_names,
73
+ input_shapes=input_shapes, name_policy=name_policy,
74
+ verbose=verbose, change_ordering=change_ordering)
75
return k_model
0 commit comments