proper epoch size

amiltonwong · Aug 30, 2016 · e98bd08 · e98bd08
1 parent 43a5217
commit e98bd08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_conv_net.py b/train_conv_net.py
@@ -155,7 +155,7 @@ def max_pool_2x2(x):
     predictors = train_predictors[data_index:data_index+50]
     target = train_targets[data_index:data_index+50]
 
-    if i%1 == 0:
+    if i%425 == 0:
 
         # Not sure what these two lines do
         run_opts = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
@@ -194,4 +194,4 @@ def max_pool_2x2(x):
 #print("validation accuracy %g" % accuracy.eval(feed_dict={x: validation_predictors, y_: validation_targets, keep_prob: 1.0}))
 
 # Marks unambiguous successful completion to prevent deletion by cleanup script
-shell_command('touch '+tfboard_run_dir+'/SUCCESS')
+shell_command('touch '+tfboard_run_dir+'/SUCCESS')