You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I followed the steps exactly as mentioned. Downloaded the multi-lang model, used the python script to get the pb file. But in Java side, I get the following when I extract embeddings.
For some reason, I must do this on TF 1.9 only.
(It works perfectly fine from TF 1.11 onwards). Any idea how can I crack this for TF 1.9?
2020-04-27 16:34:13.450298: I tensorflow/cc/saved_model/loader.cc:291] SavedModel load for tags { serve }; Status: success. Took 1703727 microseconds. Exception in thread "main" java.lang.IllegalArgumentException: logits must be 2-dimensional [[Node: bert/encoder/layer_0/attention/self/Softmax = Softmax[T=DT_FLOAT, _output_shapes=[[1,12,128,128]], _device="/job:localhost/replica:0/task:0/device:CPU:0"](bert/encoder/layer_0/attention/self/add)]]
PS: Thank you so much for this project. I just landed on the right Python script and Java module I needed the most.
The text was updated successfully, but these errors were encountered:
I followed the steps exactly as mentioned. Downloaded the multi-lang model, used the python script to get the pb file. But in Java side, I get the following when I extract embeddings.
For some reason, I must do this on TF 1.9 only.
(It works perfectly fine from TF 1.11 onwards). Any idea how can I crack this for TF 1.9?
2020-04-27 16:34:13.450298: I tensorflow/cc/saved_model/loader.cc:291] SavedModel load for tags { serve }; Status: success. Took 1703727 microseconds. Exception in thread "main" java.lang.IllegalArgumentException: logits must be 2-dimensional [[Node: bert/encoder/layer_0/attention/self/Softmax = Softmax[T=DT_FLOAT, _output_shapes=[[1,12,128,128]], _device="/job:localhost/replica:0/task:0/device:CPU:0"](bert/encoder/layer_0/attention/self/add)]]
PS: Thank you so much for this project. I just landed on the right Python script and Java module I needed the most.
The text was updated successfully, but these errors were encountered: