Decrease evaluation steps in BC

rlworkgroup · Oct 16, 2020 · 1485dc8 · 1485dc8
1 parent 6da4d52
commit 1485dc8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/garage/torch/algos/bc.py b/src/garage/torch/algos/bc.py
@@ -102,7 +102,8 @@ def train(self, trainer):
             if self._eval_env is not None:
                 log_performance(epoch,
                                 obtain_evaluation_episodes(
-                                    self.learner, self._eval_env),
+                                    self.learner, self._eval_env,
+                                    num_eps=10),
                                 discount=1.0)
             losses = self._train_once(trainer, epoch)
             with tabular.prefix(self._name + '/'):