tweak readme

lucidrains · lucidrains · commit e872ec361881 · 2022-11-21T10:08:56.000-08:00
diff --git 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()
 

Original file line number	Diff line number	Diff line change
`@@ -42,7 +42,7 @@ diffusion = GaussianDiffusion(`
`42`	`42`	`loss_type = 'l1' # L1 or L2`
`43`	`43`	`)`
`44`	`44`
`45`		`-training_images = torch.randn(8, 3, 128, 128) # images are normalized from 0 to 1`
	`45`	`+training_images = torch.rand(8, 3, 128, 128) # images are normalized from 0 to 1`
`46`	`46`	`loss = diffusion(training_images)`
`47`	`47`	`loss.backward()`
`48`	`48`	`# after a lot of training`
`@@ -124,7 +124,7 @@ diffusion = GaussianDiffusion1D(`
`124`	`124`	`objective = 'pred_v'`
`125`	`125`	`)`
`126`	`126`
`127`		`-training_seq = torch.randn(8, 32, 128) # features are normalized from 0 to 1`
	`127`	`+training_seq = torch.rand(8, 32, 128) # features are normalized from 0 to 1`
`128`	`128`	`loss = diffusion(training_seq)`
`129`	`129`	`loss.backward()`
`130`	`130`