mirror of
https://github.com/wassname/kair_algorithms_draft.git
synced 2026-10-30 13:20:08 +08:00
1 parent
61471fdf9c
commit
ede6fa3c7a
12 files changed
+243
-19
No files matched your search
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user