Fix a monitor bug

ikostrikov · Nov 7, 2017 · b8d4c2c · b8d4c2c · SerialIterator · Jan 7, 2019
1 parent 7d18622
commit b8d4c2c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -239,8 +239,11 @@ def update_current_state(state):
                        final_rewards.max(), -dist_entropy.data[0],
                        value_loss.data[0], action_loss.data[0]))
         if args.vis and j % args.vis_interval == 0:
-            win = visdom_plot(viz, win, args.log_dir, args.env_name, args.algo)
-
+            try:
+                # Sometimes monitor doesn't properly flush the outputs
+                win = visdom_plot(viz, win, args.log_dir, args.env_name, args.algo)
+            except IOError:
+                pass
 
 if __name__ == "__main__":
     main()