Merge pull request #134 from wmvanvliet/fix

Put back line that was removed by accident
gmalivenko · Aug 6, 2021 · 9cd26e9 · 9cd26e9
2 parents cf8f53f + d493cd3
commit 9cd26e9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch2keras/converter.py b/pytorch2keras/converter.py
@@ -69,4 +69,7 @@ def pytorch_to_keras(
     stream.seek(0)
     onnx_model = onnx.load(stream)
 
+    k_model = onnx_to_keras(onnx_model=onnx_model, input_names=input_names,
+                            input_shapes=input_shapes, name_policy=name_policy,
+                            verbose=verbose, change_ordering=change_ordering)
     return k_model