Browse Source

enabled progress bar in the evaluator

Dimitri Korsch 5 years ago
parent
commit
fba8601fce
1 changed files with 3 additions and 1 deletions
  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