更换训练集，训练集只有三个label，text_predict载入模型报错
lanlan-bupt opened this issue · comments
更换训练集后修改了text_model中的num_classess参数为3，同时更改了loader中的label。
训练完成后使用text_predict文件载入模型报错，
saver.restore(sess=session, save_path=save_path)这里失败。
检查过save_path没有问题。。。
请问为什么呢。。。我想不出来QAQ
我解决啦~谢谢！！
由于我更换了训练集，词文本的长度发生了变化，需要在模型中修改vocab_size的数值。
修改后运行成功。
打扰啦