KeyError: "The name 'input_word_ids:0' refers to a Tensor which does not exist. The operation, 'input_word_ids', does not exist in the graph."
dshahrokhian opened this issue · 0 comments
dshahrokhian commented
While trying to convert a RoBERTa model, tf-trt
complains about input_word_ids
not being in the graph. Nonetheless, if I run:
saved_model_cli show --dir {saved_model_dir} --all
it clearly shows the input as part of the graph:
Defined Functions:
...
...
...
Option #2
Callable with:
Argument #1
input_word_ids: TensorSpec(shape=(None, 192), dtype=tf.int32, name='input_word_ids')
Argument #2
DType: bool
Value: True
Argument #3
DType: NoneType
Value: None
...
...
...
Any ideas?