Update README.md

mlfoundations · May 8, 2024 · 2f5a8f4 · 2f5a8f4
1 parent 9eaf242
commit 2f5a8f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -50,6 +50,7 @@ from PIL import Image
 import open_clip
 
 model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-32', pretrained='laion2b_s34b_b79k')
+model.eval()  # model in train mode by default, impacts some models with BatchNorm or stochastic depth active
 tokenizer = open_clip.get_tokenizer('ViT-B-32')
 
 image = preprocess(Image.open("docs/CLIP.png")).unsqueeze(0)