From 1b6fe88bcb5ec5c7937f0fa8fffa61c805eb4472 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 19 Jul 2022 11:19:59 -0600 Subject: [PATCH] spit out overages in GDI --- 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 dc75da67..6d461ba2 100644 --- a/codes/trainer/injectors/gaussian_diffusion_injector.py +++ b/codes/trainer/injectors/gaussian_diffusion_injector.py @@ -80,7 +80,7 @@ class GaussianDiffusionInjector(Injector): def forward(self, state): gen = self.env['generators'][self.opt['generator']] hq = state[self.input] - assert hq.max() < 1 or hq.min() > -1, "Attempting to train gaussian diffusion on un-normalized inputs. This won't work, silly!" + assert hq.max() < 1 or hq.min() > -1, f"Attempting to train gaussian diffusion on un-normalized inputs. This won't work, silly! {hq.min()} {hq.max()}" with autocast(enabled=self.env['opt']['fp16']): if not gen.training or (self.deterministic_timesteps_every != 0 and self.env['step'] % self.deterministic_timesteps_every == 0):