allenai / allenai/scibert

AttributeError while using SciBERT

Offen
#88 3 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
1.7k
Forks
232
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

``
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'

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.