biodatlab
/

MIReAD-Neuro

Text Classification

text-embeddings-inference

Model card Files Files and versions

atrytone commited on Jun 9, 2023

Commit

a27dae8

·

1 Parent(s): 3c1f754

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -59,7 +59,7 @@ with torch.no_grad():
 # for classification
 output = model(**tokens)
-class = out.logits
 ```
 ## Training procedure

 # for classification
 output = model(**tokens)
+class = output.logits
 ```
 ## Training procedure