text icon indicating copy to clipboard operation
text copied to clipboard

InvalidArgumentError

Open ghost opened this issue 1 year ago • 3 comments

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 _wrapped__Pack_N_2_device/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!!!)

ghost avatar Apr 01 '24 15:04 ghost

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

devashish-bhake avatar May 29 '24 12:05 devashish-bhake

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.

ctaguchi avatar Jun 05 '24 12:06 ctaguchi