saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-26 03:01:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove stdout flush

Browse source
This commit is contained in:
Antonin Raffin 2019-09-09 13:51:18 +02:00
parent 12431b0e92
commit d333abe963
2 changed files with 0 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/cem_rl/cem_rl.py
View file

@ -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

2
torchy_baselines/td3/td3.py
View file

@ -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()