🤦 Stupid typo...

Thanks for a reddit user to point out an error in the readme.
rentruewang · Nov 29, 2021 · 8d31406 · 8d31406
1 parent 073beec
commit 8d31406
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -97,7 +97,7 @@ You change this line of code:
 # Wrap the input tensor.
 # If a batch argument is provided, that dimension of the tensor would be treated as the batch.
 # In this case, the first dimension (dim=0) is used as batch's dimension.
-input = lazy(torch.nn(8, 28, 28), batch=0)
+input = lazy(torch.randn(8, 28, 28), batch=0)
 ```
 
 Done. You will not run out of memory again.