Browse Source

enabled progress bar in the evaluator

Dimitri Korsch 5 năm trước cách đây
mục cha
commit
fba8601fce
1 tập tin đã thay đổi với 3 bổ sung1 xóa
  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(
 		self.evaluator = extensions.Evaluator(
 			iterator=self.val_iter,
 			iterator=self.val_iter,
 			target=self.clf,
 			target=self.clf,
-			device=self.device)
+			device=self.device,
+			progress_bar=True
+		)
 
 
 		self.evaluator.default_name = default_name
 		self.evaluator.default_name = default_name