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

divided by zero

Browse Source
This commit is contained in:
James Betker 2022-06-21 20:26:19 -06:00
parent f0117150d0
commit aeff1a4cc7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/trainer/injectors/gaussian_diffusion_injector.py
View File

@ -62,9 +62,9 @@ class GaussianDiffusionInjector(Injector):
def extra_metrics(self):
uqt = self.latest_timesteps > self.num_timesteps * 3 / 4
uql = (self.latest_mse_by_batch * uqt).sum() / uqt.sum()
uql = (self.latest_mse_by_batch * uqt).sum() / uqt.sum() if uqt.sum() != 0 else 0
muqt = (self.latest_timesteps > self.num_timesteps / 2) * (self.latest_timesteps < self.num_timesteps * 3 / 4)
muql = (self.latest_mse_by_batch * muqt).sum() / muqt.sum()
muql = (self.latest_mse_by_batch * muqt).sum() / muqt.sum() if muqt.sum() != 0 else 0
d = {
'upper_quantile_mse_loss': uql,
'mid_upper_quantile_mse_loss': muql,