Update DEN_run.py

jaehong31#2
yhgon · Feb 6, 2019 · 434c1e8 · 434c1e8
1 parent 0211e53
commit 434c1e8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/DEN_run.py b/DEN_run.py
@@ -10,7 +10,10 @@
 flags.DEFINE_float("lr", 0.001, "Learing rate(init) for train")
 flags.DEFINE_integer("batch_size", 256, "The size of batch for 1 iteration")
 flags.DEFINE_string("checkpoint_dir", "checkpoints", "Directory path to save the checkpoints")
-flags.DEFINE_integer("dims", [784, 312, 128, 10], "Dimensions about layers including output")
+flags.DEFINE_integer("dims0", 784, "Dimensions about input layer")
+flags.DEFINE_integer("dims1", 312, "Dimensions about 1st layer")
+flags.DEFINE_integer("dims2", 128, "Dimensions about 2nd layer")
+flags.DEFINE_integer("dims3", 10, "Dimensions about output layer")
 flags.DEFINE_integer("n_classes", 10, 'The number of classes at each task')
 flags.DEFINE_float("l1_lambda", 0.00001, "Sparsity for L1")
 flags.DEFINE_float("l2_lambda", 0.0001, "L2 lambda")