From 09a9079e17aa1e82ca2fe3491427bbb370861a7a Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 25 Aug 2020 16:00:49 -0600 Subject: [PATCH] Check rank before doing image logging. --- codes/models/SRGAN_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py index 9d59e40a..16c455cd 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -773,7 +773,7 @@ class SRGANModel(BaseModel): # Log sample images from first microbatch. - if step % self.img_debug_steps == 0: + if step % self.img_debug_steps == 0 and self.rank <= 0: sample_save_path = os.path.join(self.opt['path']['models'], "..", "temp") os.makedirs(os.path.join(sample_save_path, "hr"), exist_ok=True) os.makedirs(os.path.join(sample_save_path, "lr"), exist_ok=True)