AttributeError while using SciBERT
Aperta
- Lingua principale
- Python
- Stelle
- 1.7k
- Fork
- 232
- Metriche di merge delle PR
- Nessuna PR unita negli ultimi 30g
Descrizione
``
import tensorflow as tf
from transformers import * #BertTokenizer, TFBertModel
tokenizer = AutoTokenizer.from_pretrained('allenai/scibert_scivocab_uncased')
model = AutoModel.from_pretrained('allenai/scibert_scivocab_uncased')
input_ids = tf.constant(tokenizer.encode("My dog is cute"))[None, :] # Batch size 1
outputs = model(input_ids)
last_hidden_states = outputs[0]
print(last_hidden_states)
``
I'm getting following error:
input_shape = input_ids.size()
AttributeError: 'tensorflow.python.framework.ops.EagerTensor' object has no attribute 'size'
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Valutazione
Questa issue non è ancora stata valutata.