llm-jp
/

llm-jp-clip-vit-large-patch14

Zero-Shot Image Classification

Model card Files Files and versions Community

speed commited on Jan 2

Commit

a256ce5

·

verified ·

1 Parent(s): 6dd62a6

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -44,6 +44,7 @@ with torch.no_grad(), torch.cuda.amp.autocast():
     text_probs = (100.0 * image_features @ text_features.T).softmax(dim=-1)
 print("Label probs:", text_probs)
 ```
 Reference:

     text_probs = (100.0 * image_features @ text_features.T).softmax(dim=-1)
 print("Label probs:", text_probs)
+# Label probs: tensor([[9.9425e-01, 5.2273e-03, 5.2600e-04]])
 ```
 Reference: