Update run.py

Author: Edward-Sun

codes/run.py

@@ -284,7 +284,6 @@ def main(args):
 
     logging.info('Start Training...')
     logging.info('init_step = %d' % init_step)
-    logging.info('learning_rate = %d' % current_learning_rate)
     logging.info('batch_size = %d' % args.batch_size)
     logging.info('negative_adversarial_sampling = %d' % args.negative_adversarial_sampling)
     logging.info('hidden_dim = %d' % args.hidden_dim)
@@ -296,6 +295,8 @@ def main(args):
     # Set valid dataloader as it would be evaluated during training
 
     if args.do_train:
+        logging.info('learning_rate = %d' % current_learning_rate)
+
         training_logs = []
 
         #Training Loop
@@ -357,4 +358,4 @@ def main(args):
         log_metrics('Test', step, metrics)
 
 if __name__ == '__main__':
-    main(parse_args())
+    main(parse_args())