From 1485dc8fbca3b732a46339d2e716f58c6bae7d86 Mon Sep 17 00:00:00 2001 From: "K.R. Zentner" Date: Mon, 20 Jul 2020 17:01:24 -0700 Subject: [PATCH] Decrease evaluation steps in BC --- src/garage/torch/algos/bc.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 + '/'):