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