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

Modify docker related commands (#54)

Browse Source 
* Add new ros package installation(orocos, hrl-kdl)

* Add source bashrc

* update train script

* Add custom package ros

* Change delegating env_name logic

* Add reacher env when training selection

* Change clone branch to master
This commit is contained in:
Whi Kwon
2019-05-21 13:26:52 +09:00
committed by Seungjae Ryan Lee
parent 5ff59a280e
commit 4b56ae781b
3 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/algorithms/common/abstract/agent.py
+4 -1
View File
@@ -44,7 +44,10 @@ class AbstractAgent(object):
self.args.max_episode_steps = env._max_episode_steps
# for logging
self.env_name = str(self.env.env).split("<")[1].replace(">>", "")
if hasattr(env, "env_name"):
self.env_name = env.env_name
else:
self.env_name = self.env.unwrapped.spec.id
self.sha = (
subprocess.check_output(["git", "rev-parse", "--short", "HEAD"])[:-1]
.decode("ascii")