model saves every 15th epoch

gojibjib · Jun 20, 2018 · baca3d5 · baca3d5
1 parent 96cf793
commit baca3d5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/code/vggish_train.py b/code/vggish_train.py
@@ -362,8 +362,9 @@ def main(_):
 
       # Save model to disk.
       saver = tf.train.Saver()
-      save_path = saver.save(sess, os.path.join(model_dir, "jibjib_model.ckpt"),global_step=step)
-      log.info("Model saved to %s" % save_path)
+      if step % 15 == 0:
+        save_path = saver.save(sess, os.path.join(model_dir, "jibjib_model.ckpt"),global_step=step)
+        log.info("Model saved to %s" % save_path)
 
     now = datetime.datetime.now().isoformat().replace(":", "_").split(".")[0]
     end = time.time()