From e872ec361881dc74cfda48e3d76ff0148b63e1f4 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Mon, 21 Nov 2022 10:08:56 -0800 Subject: [PATCH] tweak readme --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 25016ed..1514ee1 100644 --- a/README.md +++ b/README.md @@ -42,7 +42,7 @@ diffusion = GaussianDiffusion( loss_type = 'l1' # L1 or L2 ) -training_images = torch.randn(8, 3, 128, 128) # images are normalized from 0 to 1 +training_images = torch.rand(8, 3, 128, 128) # images are normalized from 0 to 1 loss = diffusion(training_images) loss.backward() # after a lot of training @@ -124,7 +124,7 @@ diffusion = GaussianDiffusion1D( objective = 'pred_v' ) -training_seq = torch.randn(8, 32, 128) # features are normalized from 0 to 1 +training_seq = torch.rand(8, 32, 128) # features are normalized from 0 to 1 loss = diffusion(training_seq) loss.backward()