diff --git a/torchy_baselines/cem_rl/cem_rl.py b/torchy_baselines/cem_rl/cem_rl.py index 977224c..fccd9b3 100644 --- a/torchy_baselines/cem_rl/cem_rl.py +++ b/torchy_baselines/cem_rl/cem_rl.py @@ -1,4 +1,3 @@ -import sys import time import torch as th @@ -113,7 +112,6 @@ class CEMRL(TD3): if self.verbose > 0: print("Eval num_timesteps={}, mean_reward={:.2f}".format(self.num_timesteps, evaluations[-1])) print("FPS: {:.2f}".format(self.num_timesteps / (time.time() - start_time))) - sys.stdout.flush() actor_steps = 0 # evaluate all actors diff --git a/torchy_baselines/td3/td3.py b/torchy_baselines/td3/td3.py index 5b9fcdc..4a037cb 100644 --- a/torchy_baselines/td3/td3.py +++ b/torchy_baselines/td3/td3.py @@ -1,4 +1,3 @@ -import sys import time import torch as th @@ -178,7 +177,6 @@ class TD3(BaseRLModel): if self.verbose > 0: print("Eval num_timesteps={}, mean_reward={:.2f}".format(self.num_timesteps, evaluations[-1])) print("FPS: {:.2f}".format(self.num_timesteps / (time.time() - start_time))) - sys.stdout.flush() # Reset environment obs = self.env.reset()