Skip to content

Commit 9cd26e9

Browse files
authored
Merge pull request #134 from wmvanvliet/fix
Put back line that was removed by accident
2 parents cf8f53f + d493cd3 commit 9cd26e9

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pytorch2keras/converter.py

+3
Original file line numberDiff line numberDiff line change
@@ -69,4 +69,7 @@ def pytorch_to_keras(
6969
stream.seek(0)
7070
onnx_model = onnx.load(stream)
7171

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)
7275
return k_model

0 commit comments

Comments
 (0)