def load_args(load_key='default'):
"""get train args of retore id"""
load_data = Config.get_load_data(load_key)
if load_data is None:
return False
args_dict = load_data['args']
#Config.parse_args_dict(args_dict)
return args_dict
def create_act_model(sess, env, nenvs):
load_data = Config.get_load_data('default')
create_additional = 'use_minimum_model' not in load_data['args']
ob_space = env.observation_space
ac_space = env.action_space
policy = policies.get_policy()
act = policy(sess,
ob_space,
ac_space,
nenvs,
1,
reuse=False,
create_additional=create_additional)
return act
def load_params_for_scope(sess, scope, load_key='default'):
load_data = Config.get_load_data(load_key)
if load_data is None:
return False
params_dict = load_data['params']
if scope in params_dict:
print('Loading saved file for scope', scope)
loaded_params = params_dict[scope]
loaded_params, params = get_savable_params(loaded_params, scope, keep_heads=True)
restore_params(sess, loaded_params, params)
return True
def try_load_model(self):
load_data = Config.get_load_data('default')
if load_data is None:
return False
params_dict = load_data['params']
if "model" in params_dict:
print('Loading saved file for scope', "model")
loaded_params = params_dict["model"]
if len(loaded_params) != len(self.params):
print('param mismatch', len(loaded_params), len(self.params))
assert (False)
restore_ops = []
for p, loaded_p in zip(self.params, loaded_params):
restore_ops.append(tf.assign(p, loaded_p))
self.sess.run(restore_ops)
return True
return False
def load_params_for_scope(sess, scope, load_key='default', load_path=None):
if load_path is None:
load_data = Config.get_load_data(load_key)
else:
load_path = file_to_path(load_path)
if os.path.exists(load_path):
load_data = joblib.load(load_path)
print('Load file', load_path)
else:
load_data = None
if load_data is None:
return False
params_dict = load_data['params']
if scope in params_dict:
print('Loading saved file for scope', scope)
loaded_params = params_dict[scope]
loaded_params, params = get_savable_params(loaded_params,
scope,
keep_heads=True)
restore_params(sess, loaded_params, params)
return True
