TJMUCH
/

transcriptome-iseeek

Model card Files Files and versions

lixiangchun commited on Dec 1, 2021

Commit

794c624

·

1 Parent(s): 691a03c

update README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ batch = tokenizer(a, max_length=128, truncation=True, padding=True, return_tenso
 out = iseeek.bert(**batch)
 # [CLS] representation
-feature = last_hidden_state[:,0,:]
 ```

 out = iseeek.bert(**batch)
 # [CLS] representation
+feature = out.last_hidden_state[:,0,:]
 ```