diff --git a/eval.py b/eval.py index 30244c7b..646e5dc7 100644 --- a/eval.py +++ b/eval.py @@ -44,6 +44,7 @@ def actor(batch): GC.Start() evaluator.episode_start(0) + model.train(False) for n in eval_iters.iters(): GC.Run()