Add HER on sac (#85)

* Add HER on sac

* Add args in wandb log
This commit is contained in:
Kyunghwan Kim
2019-07-20 17:36:01 +09:00
committed by Whi Kwon
parent 61471fdf9c
commit ede6fa3c7a
12 changed files with 243 additions and 19 deletions
+1
View File
@@ -234,6 +234,7 @@ class Agent(AbstractAgent):
if self.args.log:
wandb.init()
wandb.config.update(self.hyper_params)
wandb.config.update(vars(self.args))
wandb.watch([self.actor, self.critic1, self.critic2], log="parameters")
for i_episode in range(1, self.args.episode_num + 1):