minor fix

andabi · Dec 13, 2017 · 8e25ab6 · 8e25ab6
1 parent f01f6b1
commit 8e25ab6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/convert.py b/convert.py
@@ -17,7 +17,7 @@
 from utils import denormalize_0_1
 
 
-def convert(logdir, step, writer, queue=False):
+def convert(logdir, writer, queue=False, step=None):
     hp = Hparam.get_global_hparam()
 
     # Load graph
@@ -116,7 +116,7 @@ def get_arguments():
     s = datetime.datetime.now()
 
     writer = tf.summary.FileWriter(logdir)
-    convert(logdir=logdir, step=args.step, writer=writer)
+    convert(logdir=logdir, writer=writer, step=args.step)
     writer.close()
 
     e = datetime.datetime.now()