Merge pull request #632 from mil-tokyo/dev-bin

#631 fix placeholder duplication
mil-tokyo · Nov 28, 2017 · abd516b · abd516b
2 parents 991172b + 22eab92
commit abd516b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bin/convert_keras.py b/bin/convert_keras.py
@@ -59,7 +59,7 @@ def main():
 
     model = keras.models.load_model(args.kerasmodel, custom_objects=custom_objects, compile=False)
     model.build(input_shape=None)
-    converter = KerasConverter(batch_size=Placeholder(label='N'))
+    converter = KerasConverter(batch_size=input_shapes[0][0])
     graph = converter.convert(model)
     traverse.dump(graph)