From ea20721632477a7f9da5e9ca65827391ddc06522 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Wed, 15 Jan 2020 15:58:45 +0100 Subject: [PATCH] Add TODO --- torchy_baselines/common/base_class.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py index f517c58..4c7a764 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -596,6 +596,7 @@ class BaseRLModel(object): total_episodes += 1 episode_rewards.append(episode_reward) total_timesteps.append(episode_timesteps) + # TODO: reset SDE matrix at the end of the episode? if action_noise is not None: action_noise.reset()