a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Add wandb_step_factor argument

Browse Source
This commit is contained in:
James Betker 2022-01-27 19:58:58 -07:00
parent e0e36ed98c
commit 0152174c0e
1 changed files with 1 additions and 1 deletions

2
codes/train.py
View File

@ -210,7 +210,7 @@ class Trainer:
self.tb_logger.add_scalar(k, v, self.current_step)
if opt['wandb'] and self.rank <= 0:
import wandb
wandb.log(logs, step=self.current_step)
wandb.log(logs, step=int(self.current_step * opt_get(opt, ['wandb_step_factor'], 1)))
self.logger.info(message)
#### save models and training states