probably too harsh on ninfs
This commit is contained in:
parent
4509cfc705
commit
d44ed5d12d
|
@ -697,9 +697,8 @@ class GaussianDiffusion:
|
||||||
|
|
||||||
logperp = torch.log(normpdf(err, mean, var)) / self.num_timesteps + logperp
|
logperp = torch.log(normpdf(err, mean, var)) / self.num_timesteps + logperp
|
||||||
# Remove -infs, which do happen pretty regularly (and penalize them proportionately).
|
# Remove -infs, which do happen pretty regularly (and penalize them proportionately).
|
||||||
num_infs = torch.isinf(logperp).sum()
|
logperp[torch.isinf(logperp)] = torch.max(logperp) * 2
|
||||||
logperp[torch.isinf(logperp)] = torch.max(logperp) * num_infs * 2
|
print(f'Num infs: : {torch.isinf(logperp).sum()}') # probably should just log this.
|
||||||
print(f'Num infs: : {num_infs}') # probably should just log this.
|
|
||||||
return -logperp.mean()
|
return -logperp.mean()
|
||||||
|
|
||||||
def ddim_sample(
|
def ddim_sample(
|
||||||
|
|
Loading…
Reference in New Issue
Block a user