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

fix train bug

Browse Source
This commit is contained in:
James Betker 2022-02-11 11:18:15 -07:00
parent 006add64c5
commit 4abc094b47
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

@ -291,7 +291,7 @@ class Trainer:
self.logger.info('Start training from epoch: {:d}, iter: {:d}'.format(self.start_epoch, self.current_step)) self.logger.info('Start training from epoch: {:d}, iter: {:d}'.format(self.start_epoch, self.current_step))
for epoch in range(self.start_epoch, self.total_epochs + 1): for epoch in range(self.start_epoch, self.total_epochs + 1):
self.epoch = epoch self.epoch = epoch
if opt['dist']: if self.opt['dist']:
self.train_sampler.set_epoch(epoch) self.train_sampler.set_epoch(epoch)
tq_ldr = tqdm(self.train_loader) if self.rank <= 0 else self.train_loader tq_ldr = tqdm(self.train_loader) if self.rank <= 0 else self.train_loader