search

xmxoxo / BERT-train2deploy

BERT模型从训练到部署
517 stars 165 forks source link

freeze_graph.py的问题2 #15

Closed lotest closed 3 years ago

lotest commented 4 years ago

195行的latest_checkpoint = tf.train.latest_checkpoint(args.model_dir) 参数应该是 args.bert_model_dir

fathouse commented 4 years ago

你不要乱讲啊，明明是对的

lotest commented 4 years ago

是么，我不知道哪里的问题。改了之后才能跑

fathouse commented 4 years ago

这个是classification_model.pb的生成路径，改了这个，如果不进行其他相应的修改，生成的时候会找不到存储路径

lotest commented 4 years ago

对，是需要指定训练得到的模型文件的路径