Skip to content

Commit 54cea92

Browse files
committed
Fix docstring typo of zero_shot_classifier
1 parent 5001237 commit 54cea92

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

clip_benchmark/metrics/zeroshot_classification.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@ def zero_shot_classifier(model, tokenizer, classnames, templates, device, amp=Tr
3434
Returns
3535
-------
3636
37-
torch.Tensor of shape (N,C) where N is the number
38-
of templates, and C is the number of classes.
37+
torch.Tensor of shape (D,C) where D is the projection
38+
dimension, and C is the number of classes.
3939
"""
4040
with torch.no_grad(), torch.autocast(device, enabled=amp):
4141
zeroshot_weights = []

0 commit comments

Comments
 (0)