search

tensorflow / text

Making text a first-class citizen in TensorFlow.
https://www.tensorflow.org/beta/tutorials/tensorflow_text/intro
Apache License 2.0
1.23k stars 344 forks source link

InvalidArgumentError #1265

Open ghost opened 7 months ago

ghost commented 7 months ago

I am getting an error in the decoder inference part.

InvalidArgumentError Traceback (most recent call last) in <cell line: 2>() 1 # Setup the loop variables. ----> 2 next_token, done, state = decoder.get_initial_state(ex_context) 3 tokens = [] 4 5 for n in range(10):

6 frames /usr/local/lib/python3.10/dist-packages/tensorflow/python/framework/ops.py in raise_from_not_ok_status(e, name) 5981 def raise_from_not_ok_status(e, name) -> NoReturn: 5982 e.message += (" name: " + str(name if name is not None else "")) -> 5983 raise core._status_to_exception(e) from None # pylint: disable=protected-access 5984 5985

InvalidArgumentError: {{function_node wrappedPack_N_2device/job:localhost/replica:0/task:0/device:GPU:0}} Shapes of all inputs must match: values[0].shape = [64,1,256] != values[1].shape = [] [Op:Pack] name:

Screenshot from 2024-04-01 21-46-24

Help me with this(Important!!!)

devashish-bhake commented 6 months ago

I am having the exact same error OP did you find a fix to the issue?

ctaguchi commented 5 months ago

The same error happened to me with my code that was working a few month ago. In my case, downgrading tensorflow-text to an older version solved it, like pip install "tensorflow-text==2.11.