dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed replay

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-29 17:02:40 +02:00
parent 28561b9bb2
commit 30c9e93967
1 changed files with 9 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
replay.py
View File

@ -27,23 +27,13 @@ def main(load_path, n_eval_episodes=0):
print(env_name, alg_name, alg_deriv, use_sde) print(env_name, alg_name, alg_deriv, use_sde)
env = gym.make(env_name) env = gym.make(env_name)
if alg_name == 'ppo': if alg_name == 'ppo':
model = PPO( Model = PPO
"MlpPolicy",
env,
use_sde=use_sde,
)
elif alg_name == 'trl' and alg_deriv == 'pg': elif alg_name == 'trl' and alg_deriv == 'pg':
model = TRL_PG( Model = TRL_PG
"MlpPolicy",
env,
use_sde=use_sde,
)
else: else:
raise Exception('Algorithm not implemented for replay') raise Exception('Algorithm not implemented for replay')
print(model.get_parameters()) model = Model.load(load_path, env=env)
model.load(load_path, env=env)
if n_eval_episodes: if n_eval_episodes:
mean_reward, std_reward = evaluate_policy( mean_reward, std_reward = evaluate_policy(