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

add grad scaler scale to metrics

Browse Source
This commit is contained in:
James Betker 2022-03-08 15:52:42 -07:00
parent d2bdeb6f20
commit 3e5da71b16
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/trainer/steps.py
View File

@ -344,4 +344,6 @@ class ConfigurableStep(Module):
opt.zero_grad() opt.zero_grad()
def get_metrics(self): def get_metrics(self):
return self.loss_accumulator.as_dict() metrics = self.loss_accumulator.as_dict()
metrics['grad_scaler_scale'] = self.scaler.get_scale()
return metrics