mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki 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:
wandb_logs[k] = v
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:
wandb.log(logs, step=self.total_training_data_encountered)
wandb.log(wandb_logs, step=self.total_training_data_encountered)
self.logger.info(message)
#### save models and training states