models icon indicating copy to clipboard operation
models copied to clipboard

Missing tensor.name for all the input/output pb files in bertsquad-8.tar.gz

Open winnietsang opened this issue 5 years ago • 0 comments

In text/machine_comprehension/bert-squad/model/bertsquad-8.tar.gz, there are input and output pb files. Base on the README.md of BERT-Squad model, there should have 4 inputs and 3 outputs, without the tensor.name defined in the pb files, it is very hard to determine which input is which and the same apply to output too. Further, there are 4 output pb files, output_3.pb should be deleted.

Filename tensor.name
input_0.pb unique_ids_raw_output___9:0
input_1.pb input_ids:0
input_2.pb input_mask:0
input_3.pb segment_ids:0
output_0.pb unique_ids:0
output_1.pb unstack:0
output_2.pb unstack:1
~~output_3.pb~~ There are only 3 outputs, this file should be deleted.

winnietsang avatar Dec 16 '20 01:12 winnietsang