fix savor bug

a170811 · Jan 5, 2019 · f4d72ad · f4d72ad
1 parent b7ef685
commit f4d72ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/word_embed.py b/word_embed.py
@@ -136,7 +136,7 @@ def _cos_sim(a, b):
         loss = tf.reduce_mean(tf.log(1 + tf.exp(-args.temperature * gradient)))
         optimizer = tf.train.AdamOptimizer(args.learning_rate).minimize(loss)
 
-    saver = tf.train.Saver()
+        saver = tf.train.Saver()
     with tf.Session(graph = graph, config = tf.ConfigProto(log_device_placement = True)) as sess:
         #sess = tf_debug.LocalCLIDebugWrapperSession(sess)
         #sess.add_tensor_filter("has_inf_or_nan", tf_debug.has_inf_or_nan)
@@ -165,7 +165,7 @@ def _cos_sim(a, b):
             for batch_data_f, batch_data_c, neg_data in next_batch(args.mb, args.n_negs,
                     dict_size):
 
-                log_step = 20
+                log_step = 2000
                 if step % log_step == 0:
                     average_loss /= log_step
                     print('Average loss at step {} is {:.3f}'.format(step, average_loss))