removed the eval callback

2 years ago · 5587144815
1 changed files with 1 additions and 1 deletions
--- 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)}")