wassname/kair_algorithms_draft
1
0
Fork 0
mirror of https://github.com/wassname/kair_algorithms_draft.git synced 2026-09-09 11:25:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add demo files (#23)

Browse Source 
* Add demo files

* Add demo path on run files and change formatting

* Delete discrete demo
This commit is contained in:
Kyunghwan Kim
2019-03-14 14:02:24 +09:00
committed by GitHub
parent ca5c99bc41
commit b4f8bd58b4
4 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/algorithms/td3/agent.py
+4 -3
View File
@@ -65,9 +65,10 @@ class Agent(AbstractAgent):
"""
AbstractAgent.__init__(self, env, args)
self.actor, self.actor_target, self.critic1, self.critic1_target, self.critic2, self.critic2_target = ( # noqa: B950
models
)
self.actor, self.actor_target = models[:2]
self.critic1, self.critic1_target = models[2:4]
self.critic2, self.critic2_target = models[4:]
self.actor_optim, self.critic_optim = optims
self.hyper_params = hyper_params
self.exploration_noise, self.target_policy_noise = noises