|
@ -88,7 +88,7 @@ def main(): |
|
|
|
|
|
|
|
|
imageAndVideoCallback = ImageRecorderCallback(eval_env, render_freq, n_eval_episodes=1, evaluation_method=evaluate_policy, log_dir=log_dir, deterministic=True, verbose=0) |
|
|
imageAndVideoCallback = ImageRecorderCallback(eval_env, render_freq, n_eval_episodes=1, evaluation_method=evaluate_policy, log_dir=log_dir, deterministic=True, verbose=0) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
assert(False) |
|
|
model.learn(steps,callback=[imageAndVideoCallback, InfoCallback(), evalCallback]) |
|
|
model.learn(steps,callback=[imageAndVideoCallback, InfoCallback(), evalCallback]) |
|
|
model.save(f"{log_dir}/{expname(args)}") |
|
|
model.save(f"{log_dir}/{expname(args)}") |
|
|
|
|
|
|
|
|