From 38a00f29c0bb295beb4b3c960b42c461a365ff70 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Fri, 10 Jun 2022 15:41:39 -0600
Subject: [PATCH] now theres deprecation warnings, fml

---
 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 6be43bf3..d20678bc 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': torch.tensor(self.schedule_sampler._warmed_up()).float()
+                'sampler_warmed_up': torch.tensor(float(self.schedule_sampler._warmed_up()))
             }
         return {}