From 7f32a4bec7f550f8d06c052ad35c89ff9adb7da5 Mon Sep 17 00:00:00 2001 From: Nicolas Carion Date: Tue, 3 Jul 2018 15:28:18 +0200 Subject: [PATCH] Fix evaluation script --- eval.py | 1 + 1 file changed, 1 insertion(+) 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()