Browse Source

removed the eval callback

refactoring
sp 7 months ago
parent
commit
5587144815
  1. 2
      examples/shields/rl/13_minigridsb.py

2
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)}")

Loading…
Cancel
Save