From ff7c4d24f459f830af2b8d0e03b5251e212165ad Mon Sep 17 00:00:00 2001 From: Noah Dormann Date: Thu, 5 Dec 2019 08:46:26 +0100 Subject: [PATCH] deleted types in train_actor td3 --- torchy_baselines/td3/td3.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchy_baselines/td3/td3.py b/torchy_baselines/td3/td3.py index 557e901..645ecf2 100644 --- a/torchy_baselines/td3/td3.py +++ b/torchy_baselines/td3/td3.py @@ -149,9 +149,9 @@ class TD3(BaseRLModel): for param, target_param in zip(self.critic.parameters(), self.critic_target.parameters()): target_param.data.copy_(tau * param.data + (1 - tau) * target_param.data) - def train_actor(self, gradient_steps: object = 1, batch_size: object = 100, tau_actor: object = 0.005, - tau_critic: object = 0.005, - replay_data: object = None) -> object: + def train_actor(self, gradient_steps=1, batch_size=100, tau_actor=0.005, + tau_critic=0.005, + replay_data=None): # Update optimizer learning rate self._update_learning_rate(self.actor.optimizer)