CNN

astorfi · Jun 15, 2017 · d1cf6c7 · d1cf6c7
1 parent 18b0443
commit d1cf6c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/codes/3-neural_networks/convolutional-neural-network/code/train_classifier.py b/codes/3-neural_networks/convolutional-neural-network/code/train_classifier.py
@@ -29,7 +29,7 @@
 tf.app.flags.DEFINE_integer('batch_size', np.power(2, 9),
                             'Number of model clones to deploy.')
 
-tf.app.flags.DEFINE_integer('num_epochs', 3,
+tf.app.flags.DEFINE_integer('num_epochs', 1,
                             'Number of epochs for training.')
 
 ##########################################