|
@@ -70,5 +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
|
|
|
|
|
|
|
+ opts.no_progress = opts.no_progress or not self.mpi_main_process
|
|
|
super(MPIFinetuner, self).run(trainer_cls, opts, *args, **kwargs)
|
|
super(MPIFinetuner, self).run(trainer_cls, opts, *args, **kwargs)
|
