changed order of load / train modes

uclnlp · Jun 8, 2017 · 3f9e2fd · 3f9e2fd
1 parent 0b32d21
commit 3f9e2fd
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/pred.py b/pred.py
@@ -19,7 +19,7 @@
 import tensorflow as tf
 
 
-# Set mode
+# Prompt for mode
 mode = input('mode (load / train)? ')
 
 
@@ -84,6 +84,17 @@
 predict = tf.arg_max(softmaxed_logits, 1)
 
 
+# Load model
+if mode == 'load':
+    with tf.Session() as sess:
+        load_model(sess)
+
+
+        # Predict
+        test_feed_dict = {features_pl: test_set, keep_prob_pl: 1.0}
+        test_pred = sess.run(predict, feed_dict=test_feed_dict)
+
+
 # Train model
 if mode == 'train':
 
@@ -116,16 +127,5 @@
         test_pred = sess.run(predict, feed_dict=test_feed_dict)
 
 
-# Load model
-if mode == 'load':
-    with tf.Session() as sess:
-        load_model(sess)
-
-
-        # Predict
-        test_feed_dict = {features_pl: test_set, keep_prob_pl: 1.0}
-        test_pred = sess.run(predict, feed_dict=test_feed_dict)
-
-
 # Save predictions
 save_predictions(test_pred, file_predictions)