Эх сурвалжийг харах

enabled progress bar in the evaluator

Dimitri Korsch 5 жил өмнө
parent
commit
fba8601fce

+ 3 - 1
cvfinetune/finetuner/base.py

@@ -293,7 +293,9 @@ class _TrainerMixin(abc.ABC):
 		self.evaluator = extensions.Evaluator(
 			iterator=self.val_iter,
 			target=self.clf,
-			device=self.device)
+			device=self.device,
+			progress_bar=True
+		)
 
 		self.evaluator.default_name = default_name