saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-02 03:55:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

deleted types in train_actor td3

Browse source
This commit is contained in:
Noah Dormann 2019-12-05 08:46:26 +01:00
parent 4c0f6cbe53
commit ff7c4d24f4
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
torchy_baselines/td3/td3.py
View file

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