saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-16 22:20:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add TODO

Browse source
This commit is contained in:
Antonin Raffin 2020-01-15 15:58:45 +01:00
parent 03e853997a
commit ea20721632
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
torchy_baselines/common/base_class.py
View file

@ -596,6 +596,7 @@ class BaseRLModel(object):
total_episodes += 1 total_episodes += 1
episode_rewards.append(episode_reward) episode_rewards.append(episode_reward)
total_timesteps.append(episode_timesteps) total_timesteps.append(episode_timesteps)
# TODO: reset SDE matrix at the end of the episode?
if action_noise is not None: if action_noise is not None:
action_noise.reset() action_noise.reset()