Modify TensorBoard example of summit 2017

Add condition which decides to take two conv layers or one
leejaymin · Mar 20, 2017 · 1f78663 · 1f78663
1 parent b3ccbe0
commit 1f78663
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 5 deletions.
diff --git a/.idea/deployment.xml b/.idea/deployment.xml
diff --git a/7.TensorBoard/MNIST_DATA/data/t10k-images-idx3-ubyte.gz b/7.TensorBoard/MNIST_DATA/data/t10k-images-idx3-ubyte.gz
diff --git a/7.TensorBoard/MNIST_DATA/data/t10k-labels-idx1-ubyte.gz b/7.TensorBoard/MNIST_DATA/data/t10k-labels-idx1-ubyte.gz
diff --git a/7.TensorBoard/MNIST_DATA/data/train-images-idx3-ubyte.gz b/7.TensorBoard/MNIST_DATA/data/train-images-idx3-ubyte.gz
diff --git a/7.TensorBoard/MNIST_DATA/data/train-labels-idx1-ubyte.gz b/7.TensorBoard/MNIST_DATA/data/train-labels-idx1-ubyte.gz
diff --git a/7.TensorBoard/mnist.py b/7.TensorBoard/mnist.py
@@ -26,7 +26,6 @@
 urlretrieve(GIST_URL + 'labels_1024.tsv', LOGDIR + 'labels_1024.tsv')
 urlretrieve(GIST_URL + 'sprite_1024.png', LOGDIR + 'sprite_1024.png')
 
-
 def conv_layer(input, size_in, size_out, name="conv"):
   with tf.name_scope(name):
     w = tf.Variable(tf.truncated_normal([5, 5, size_in, size_out], stddev=0.1), name="W")
@@ -133,14 +132,15 @@ def main():
   # You can try adding some more learning rates
   for learning_rate in [1E-3, 1E-4, 1E-5]:
     # Include "False" as a value to try different model architectures
-    for use_two_fc in [True]:
-      for use_two_conv in [True]:
+    for use_two_fc in [True, False]:
+      for use_two_conv in [True, False]:
         # Construct a hyperparameter string for each one (example: "lr_1E-3,fc=2,conv=2)
         hparam = make_hparam_string(learning_rate, use_two_fc, use_two_conv)
         print('Starting run for %s' % hparam)
 
 	    # Actually run with the new settings
         mnist_model(learning_rate, use_two_fc, use_two_conv, hparam)
 
+
 if __name__ == '__main__':
   main()