deepmimic fix: reference initialization wasn't spread over all frames

This commit is contained in:
Erwin Coumans
2019-02-01 17:18:13 -08:00
parent 7cc2685faf
commit 1cef924973
5 changed files with 62 additions and 7 deletions

View File

@@ -3,6 +3,7 @@ import learning.agent_builder as AgentBuilder
import learning.tf_util as TFUtil
from learning.rl_agent import RLAgent
from pybullet_utils.logger import Logger
import pybullet_data
class RLWorld(object):
def __init__(self, env, arg_parser):
@@ -79,7 +80,7 @@ class RLWorld(object):
if (len(model_files) > 0):
curr_model_file = model_files[i]
if curr_model_file != 'none':
curr_agent.load_model(curr_model_file)
curr_agent.load_model(pybullet_data.getDataPath()+"/"+curr_model_file)
self.agents.append(curr_agent)
Logger.print2('')
@@ -139,4 +140,4 @@ class RLWorld(object):
assert (agent != None), 'Failed to build agent {:d} from: {}'.format(id, agent_file)
return agent