Fixed wrong element selection from layers dictionary. Closes #16

Author: gmalivenko

pytorch2keras/layers.py

@@ -20,7 +20,7 @@ def convert_conv(params, w_name, scope_name, inputs, layers, weights):
     tf_name = w_name + str(random.random())
     bias_name = '{0}.bias'.format(w_name)
     weights_name = '{0}.weight'.format(w_name)
-    input_name = layers[inputs[0]]
+    input_name = inputs[0]
 
     if len(weights[weights_name].numpy().shape) == 4:
         W = weights[weights_name].numpy().transpose(2, 3, 1, 0)
@@ -39,7 +39,7 @@ def convert_conv(params, w_name, scope_name, inputs, layers, weights):
                 padding=(params['pads'][0], params['pads'][1]),
                 name=padding_name
             )
-            layers[padding_name] = padding_layer(layers[inputs[0]])
+            layers[padding_name] = padding_layer(layers[input_name])
             input_name = padding_name
 
         weights = None