Browse Source

minor printing fix

Dimitri Korsch 6 years ago
parent
commit
5768abcf35
1 changed files with 1 additions and 0 deletions
  1. 1 0
      cvfinetune/finetuner/mpi.py

+ 1 - 0
cvfinetune/finetuner/mpi.py

@@ -70,4 +70,5 @@ class MPIFinetuner(DefaultFinetuner, _mpi_mixin):
 	def run(self, trainer_cls, opts, *args, **kwargs):
 	def run(self, trainer_cls, opts, *args, **kwargs):
 		kwargs["no_observe"] = not self.mpi_main_process
 		kwargs["no_observe"] = not self.mpi_main_process
 		opts.no_snapshot = not self.mpi_main_process
 		opts.no_snapshot = not self.mpi_main_process
+		opts.no_progress = not self.mpi_main_process
 		super(MPIFinetuner, self).run(trainer_cls, opts, *args, **kwargs)
 		super(MPIFinetuner, self).run(trainer_cls, opts, *args, **kwargs)