We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3c11564 commit 17ac96aCopy full SHA for 17ac96a
pytorch2keras/layers.py
@@ -20,7 +20,7 @@ def convert_conv(params, w_name, scope_name, inputs, layers, weights):
20
tf_name = w_name + str(random.random())
21
bias_name = '{0}.bias'.format(w_name)
22
weights_name = '{0}.weight'.format(w_name)
23
- input_name = layers[inputs[0]
+ input_name = layers[inputs[0]]
24
25
if len(weights[weights_name].numpy().shape) == 4:
26
W = weights[weights_name].numpy().transpose(2, 3, 1, 0)
0 commit comments