From 81f8e3aca7989c077b74c7ffe7ae2360114c727c Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Mon, 8 Jan 2024 17:07:58 -0700 Subject: [PATCH] Update README.md 2nd example fix, #3 --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 4b4bea1..117a06e 100644 --- a/README.md +++ b/README.md @@ -67,8 +67,9 @@ import torch # Import the torch library # Import the MMM model from the mm_mamba module from mm_mamba.model import MMM -# Generate a random tensor 'x' of size (1, 224) with random elements between 0 and 10000 -x = torch.randint(0, 10000, (1, 224)) +# Generate a random tensor 'x' of size (1, 196) with random elements between 0 and 10000 +# 196 is (224//patchsize16)**2 +x = torch.randint(0, 10000, (1, 196)) # Generate a random image tensor 'img' of size (1, 3, 224, 224) img = torch.randn(1, 3, 224, 224)