mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Check rank before doing image logging.

Browse Source
This commit is contained in:
James Betker 2020-08-25 16:00:49 -06:00
parent a1800f45ef
commit 09a9079e17
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/SRGAN_model.py
View File

@ -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)