xmxoxo / BERT-train2deploy

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

freeze_graph.py错误 #10

Open w-wenjie opened 5 years ago

w-wenjie commented 5 years ago

optimize_class_model方法里调用create_classification_model时会传入num_labels,但是这个num_labels在前面没定义。 #############################################################

增加 从label2id.pkl中读取num_labels, 这样也可以不用指定num_labels参数; 2019/4/17

    if not args.num_labels:
        num_labels, label2id, id2label = init_predict_var(tmp_dir)

############################################################# 如果执行脚本时输入了num_labels参数,则上面这段代码就不会执行,这就导致没有定义num_labels变量。

zyj108 commented 5 years ago

将183行代码的参数num_labels=num_labels改为num_labels=args.num_labels即可。