Merge pull request #10 from fgvbrt/master

fix eval script
for-ai · Jun 17, 2019 · 87cf2a0 · 87cf2a0
2 parents e84971d + d8f299e
commit 87cf2a0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/scripts/prune/eval.py b/scripts/prune/eval.py
@@ -20,16 +20,14 @@ def init_flags():
   tf.flags.DEFINE_string("data_dir", None, "The data directory.")
   tf.flags.DEFINE_integer("train_steps", 10000,
                           "Number of training steps to perform.")
-  tf.flags.DEFINE_integer("eval_steps", 100,
-                          "Number of evaluation steps to perform.")
   tf.flags.DEFINE_integer("eval_every", 1000,
                           "Number of steps between evaluations.")
   tf.flags.DEFINE_string(
       "post_weights_dir", "",
       "folder of the weights, if not set defaults to output_dir")
   tf.flags.DEFINE_string("prune_percent", "0.5",
                          "percent of weights to prune, comma separated")
-  tf.flags.DEFINE_string("prune", "one_shot", "one_shot or fisher")
+  tf.flags.DEFINE_string("prune", "weight", "one_shot or fisher")
   tf.flags.DEFINE_boolean("variational", False, "use evaluate")
   tf.flags.DEFINE_string("eval_file", "eval_prune_results",
                          "file to put results")