소스 검색

fixed logging/reporting for non-main MPI processes

Dimitri Korsch 3 년 전
부모
커밋
09aace1dae
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      cvfinetune/finetuner/mpi.py

+ 1 - 0
cvfinetune/finetuner/mpi.py

@@ -81,6 +81,7 @@ class MPIFinetuner(DefaultFinetuner):
 
 	def run(self, trainer_cls, opts, *args, **kwargs):
 		if not self.mpi_main_process:
+			kwargs["no_observe"] = True
 			opts.no_snapshot = True
 			opts.no_progress = True
 			self.evaluator._progress_bar = False