normalize inputs for much better accuracy

amiltonwong · Aug 30, 2016 · 2d53ccd · 2d53ccd
1 parent e98bd08
commit 2d53ccd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train_conv_net.py b/train_conv_net.py
@@ -138,7 +138,7 @@ def max_pool_2x2(x):
 train_writer = tf.train.SummaryWriter(train_dir,sess.graph)
 validation_writer = tf.train.SummaryWriter(validation_dir,sess.graph)
 
-
+validation_predictors[:200] = validation_predictors[:200] / 255
 
 sess.run(tf.initialize_all_variables())
 batch_index = 0
@@ -154,6 +154,7 @@ def max_pool_2x2(x):
     data_index = batch_index * 50
     predictors = train_predictors[data_index:data_index+50]
     target = train_targets[data_index:data_index+50]
+    predictors = predictors / 255
 
     if i%425 == 0: