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

damn this sucks

Browse Source
This commit is contained in:
James Betker 2022-06-10 15:38:59 -06:00
parent 0316063e2d
commit 561a6b8ff7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/trainer/injectors/gaussian_diffusion_injector.py
View File

@ -59,7 +59,7 @@ class GaussianDiffusionInjector(Injector):
def extra_metrics(self): def extra_metrics(self):
if hasattr(self, 'schedule_sampler') and isinstance(self.schedule_sampler, LossSecondMomentResampler): if hasattr(self, 'schedule_sampler') and isinstance(self.schedule_sampler, LossSecondMomentResampler):
return { return {
'sampler_warmed_up': self.schedule_sampler._warmed_up() 'sampler_warmed_up': torch.tensor(self.schedule_sampler._warmed_up()).float()
} }
return {} return {}