|
|
@@ -145,11 +145,14 @@ class Trainer(T):
|
|
|
)
|
|
|
|
|
|
|
|
|
- def setup_evaluator(self, evaluator: chainer.training.Evaluator, trigger):
|
|
|
+ def setup_evaluator(self,
|
|
|
+ evaluator: chainer.training.extensions.Evaluator,
|
|
|
+ trigger: Tuple[int, str]):
|
|
|
+
|
|
|
self.evaluator = evaluator
|
|
|
if evaluator is None:
|
|
|
return
|
|
|
- self.extend(evaluator, trigger=intervals.eval)
|
|
|
+ self.extend(evaluator, trigger=trigger)
|
|
|
|
|
|
def setup_snapshots(self, opts, obj, trigger):
|
|
|
|
