Dimitri Korsch пре 5 година
родитељ
комит
d20b137536
1 измењених фајлова са 5 додато и 2 уклоњено
  1. 5 2
      cvfinetune/training/trainer/base.py

+ 5 - 2
cvfinetune/training/trainer/base.py

@@ -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):