feedback from PR

tflearn · May 19, 2016 · 9046ecf · 9046ecf
1 parent bae9b19
commit 9046ecf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/examples/images/convnet_highway_mnist.py b/examples/images/convnet_highway_mnist.py
@@ -34,7 +34,6 @@
         network = highway_conv_2d(network, 16, j, activation='elu')
     network = max_pool_2d(network, 2)
     network = batch_normalization(network)
-    print(network)
 
 network = fully_connected(network, 128, activation='elu')
 network = fully_connected(network, 256, activation='elu')

diff --git a/examples/images/convnet_mnist.py b/examples/images/convnet_mnist.py
@@ -45,5 +45,5 @@
 # Training
 model = tflearn.DNN(network, tensorboard_verbose=0)
 model.fit({'input': X}, {'target': Y}, n_epoch=20,
-          validation_set=({'input': testX}, {'target': testY}),
-          show_metric=True, run_id='convnet_mnist')
+           validation_set=({'input': testX}, {'target': testY}),
+-          snapshot_step=100,  show_metric=True, run_id='convnet_mnist'))