Failed to read [E:\models\roBert\graph.db]: NodeDef mentions attr 'batch_dims' not in Op<name=GatherV2; signature=params:Tparams, indices:Tindices, axis:Taxis -> output:Tparams; attr=Tparams:type; attr=Tindices:type,allowed=[DT_INT32, DT_INT64]; attr=Taxis:type,allowed=[DT_INT32, DT_INT64]>; NodeDef: {{node bert/embeddings/GatherV2}} = GatherV2[Taxis=DT_INT32, Tindices=DT_INT32, Tparams=DT_FLOAT, batch_dims=0](bert/embeddings/word_embeddings/read, bert/embeddings/Reshape, bert/embeddings/GatherV2/axis). (Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary.).
java加载tf模型会抱上面的问题,一般是tf的版本不对。