From 561a6b8ff7f7726a165664073583c950c9dcfcd2 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Fri, 10 Jun 2022 15:38:59 -0600
Subject: [PATCH] damn this sucks

---
 codes/trainer/injectors/gaussian_diffusion_injector.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codes/trainer/injectors/gaussian_diffusion_injector.py b/codes/trainer/injectors/gaussian_diffusion_injector.py
index 9b2e794d..6be43bf3 100644
--- 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 {}