Explorar o código

fixed some naming bugs

Dimitri Korsch %!s(int64=5) %!d(string=hai) anos
pai
achega
d20b137536
Modificáronse 1 ficheiros con 5 adicións e 2 borrados
  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):