Skip to content

Commit b7fc5c0

Browse files
committed
Fixed error in reshape layer.
1 parent 8b1bd7a commit b7fc5c0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch2keras/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -565,7 +565,7 @@ def convert_reshape(params, w_name, scope_name, inputs, layers, weights):
565565
print('Converting reshape ...')
566566

567567
tf_name = w_name + str(random.random())
568-
reshape = keras.layers.Reshape(params['shape'], name=tf_name)
568+
reshape = keras.layers.Reshape(params['shape'][1:], name=tf_name)
569569
layers[scope_name] = reshape(layers[inputs[0]])
570570

571571

0 commit comments

Comments
 (0)