fix comment

2026-09-09 11:21:11 +08:00 · 2022-06-06 17:30:46 -07:00
parent a291da5098
commit 6b93fa48f6
1 changed files with 1 additions and 1 deletions
@@ -34,7 +34,7 @@ diffusion = GaussianDiffusion(
    loss_type = 'l1'    # L1 or L2
 )

-training_images = torch.randn(8, 3, 128, 128) # your images need to be normalized from a range of -1 to +1
+training_images = torch.randn(8, 3, 128, 128) # images are normalized from 0 to 1
 loss = diffusion(training_images)
 loss.backward()
 # after a lot of training