damn this sucks

2022-06-10 15:38:59 -06:00 · 2022-06-10 15:38:59 -06:00 · 561a6b8ff7
commit 561a6b8ff7
parent 0316063e2d
1 changed files with 1 additions and 1 deletions
--- a/codes/trainer/injectors/gaussian_diffusion_injector.py
+++ b/codes/trainer/injectors/gaussian_diffusion_injector.py
@ -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': self.schedule_sampler._warmed_up()
+                'sampler_warmed_up': torch.tensor(self.schedule_sampler._warmed_up()).float()
            }
        return {}