Quellcode durchsuchen

enabled progress bar in the evaluator

Dimitri Korsch vor 4 Jahren
Ursprung
Commit
fba8601fce
1 geänderte Dateien mit 3 neuen und 1 gelöschten Zeilen
  1. 3 1
      cvfinetune/finetuner/base.py

+ 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