deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't use tqdm in ranks!=0

Browse Source
This commit is contained in:
James Betker 2021-12-28 10:06:54 -07:00
parent a12042ea99
commit 93624fa4b2
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/train.py
View File

@ -264,7 +264,7 @@ class Trainer:
self.epoch = epoch self.epoch = epoch
if opt['dist']: if opt['dist']:
self.train_sampler.set_epoch(epoch) self.train_sampler.set_epoch(epoch)
tq_ldr = tqdm(self.train_loader) tq_ldr = tqdm(self.train_loader) if self.rank == 0 else self.train_loader
_t = time() _t = time()
for train_data in tq_ldr: for train_data in tq_ldr: