Add HER on sac (#85)

* Add HER on sac

* Add args in wandb log
This commit is contained in:
Kyunghwan Kim authored and Whi Kwon committed 2019-07-20 17:36:01 +09:00
1 parent 61471fdf9c
commit ede6fa3c7a
12 files changed
+243 -19

No files matched your search

+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):