diff --git a/src/garage/torch/algos/bc.py b/src/garage/torch/algos/bc.py index d070e21f3f..b6ec61b9f7 100644 --- 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 + '/'):