mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

now theres deprecation warnings, fml

Browse Source
This commit is contained in:
James Betker 2022-06-10 15:41:39 -06:00
parent 561a6b8ff7
commit 38a00f29c0
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):
if hasattr(self, 'schedule_sampler') and isinstance(self.schedule_sampler, LossSecondMomentResampler):
return {
'sampler_warmed_up': torch.tensor(self.schedule_sampler._warmed_up()).float()
'sampler_warmed_up': torch.tensor(float(self.schedule_sampler._warmed_up()))
}
return {}