kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

facepalm

Browse Source
This commit is contained in:
James Betker 2022-05-30 16:48:18 -06:00
parent 479c3195f3
commit 96da10415e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/train.py
View File

@ -245,9 +245,9 @@ class Trainer:
else: else:
wandb_logs[k] = v wandb_logs[k] = v
if opt_get(opt, ['wandb_progress_use_raw_steps'], False): if opt_get(opt, ['wandb_progress_use_raw_steps'], False):
wandb.log(logs, step=self.current_step) wandb.log(wandb_logs, step=self.current_step)
else: else:
wandb.log(logs, step=self.total_training_data_encountered) wandb.log(wandb_logs, step=self.total_training_data_encountered)
self.logger.info(message) self.logger.info(message)
#### save models and training states #### save models and training states