Add wandb_step_factor argument

2022-01-27 19:58:58 -07:00 · 2022-01-27 19:58:58 -07:00 · 0152174c0e
commit 0152174c0e
parent e0e36ed98c
1 changed files with 1 additions and 1 deletions
--- a/codes/train.py
+++ b/codes/train.py
@ -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