Update README.md (#283)

`patch_size` is size of patches, not number of patches
lucidrains · Oct 9, 2023 · 9e1e824 · 9e1e824
1 parent bbb24e3
commit 9e1e824
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -92,7 +92,7 @@ preds = v(img) # (1, 1000)
 - `image_size`: int.  
 Image size. If you have rectangular images, make sure your image size is the maximum of the width and height
 - `patch_size`: int.  
-Number of patches. `image_size` must be divisible by `patch_size`.  
+Size of patches. `image_size` must be divisible by `patch_size`.  
 The number of patches is: ` n = (image_size // patch_size) ** 2` and `n` **must be greater than 16**.
 - `num_classes`: int.  
 Number of classes to classify.