Base commit

examples/A3C-Gym/train-atari.py

@@ -281,7 +281,12 @@ if __name__ == '__main__':
                         choices=['play', 'eval', 'train', 'dump_video'], default='train')
     parser.add_argument('--output', help='output directory for logs and videos')
     parser.add_argument('--episode', help='number of episode to eval', default=100, type=int)
+    parser.add_argument('--render', help='Display real-time video of game-playing', action='store_true')
+
     args = parser.parse_args()
+
+    print(args.render)
+
     if args.output is None:
         args.output = os.path.join('train_log', 'train-atari-{}'.format(args.env))
 
@@ -301,7 +306,7 @@ if __name__ == '__main__':
             output_names=['policy']))
         if args.task == 'play':
             play_n_episodes(get_player(train=False), pred,
-                            args.episode, render=True)
+                            args.episode, render=args.render)
         elif args.task == 'eval':
             eval_model_multithread(pred, args.episode, get_player)
         elif args.task == 'dump_video':