Fix typo in ddpm.py

Paramterization -> Parameterization
This commit is contained in:
Ikko Ashimine 2022-11-29 11:52:30 +09:00 committed by GitHub
parent 47b6b607fd
commit d4e8234adf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -390,7 +390,7 @@ class DDPM(pl.LightningModule):
elif self.parameterization == "v":
target = self.get_v(x_start, noise, t)
else:
raise NotImplementedError(f"Paramterization {self.parameterization} not yet supported")
raise NotImplementedError(f"Parameterization {self.parameterization} not yet supported")
loss = self.get_loss(model_out, target, mean=False).mean(dim=[1, 2, 3])