From 55871448155fff72f134355452624b922ba03332 Mon Sep 17 00:00:00 2001 From: sp Date: Thu, 28 Mar 2024 14:18:42 +0100 Subject: [PATCH] removed the eval callback --- examples/shields/rl/13_minigridsb.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/shields/rl/13_minigridsb.py b/examples/shields/rl/13_minigridsb.py index 7c36bc7..616e48c 100644 --- a/examples/shields/rl/13_minigridsb.py +++ b/examples/shields/rl/13_minigridsb.py @@ -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) - model.learn(steps,callback=[imageAndVideoCallback, InfoCallback(), evalCallback]) + model.learn(steps,callback=[imageAndVideoCallback, InfoCallback()]) model.save(f"{log_dir}/{expname(args)}")