models
models copied to clipboard
Missing tensor.name for all the input/output pb files in bertsquad-8.tar.gz
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. |