Bladeren bron

fixed evaluator progress bar for the case if no_progress was set

Dimitri Korsch 4 jaren geleden
bovenliggende
commit
e124478fa1
1 gewijzigde bestanden met toevoegingen van 3 en 0 verwijderingen
  1. 3 0
      cvfinetune/training/trainer/base.py

+ 3 - 0
cvfinetune/training/trainer/base.py

@@ -85,6 +85,9 @@ class Trainer(T):
 		if not opts.no_progress:
 		if not opts.no_progress:
 			self.extend(extensions.ProgressBar(update_interval=1))
 			self.extend(extensions.ProgressBar(update_interval=1))
 
 
+		elif self.evaluator is not None:
+			self.evaluator._progress_bar = False
+
 	@property
 	@property
 	def optimizer(self):
 	def optimizer(self):
 		return self.updater.get_optimizer("main")
 		return self.updater.get_optimizer("main")