search

wzzzd / text_classifier_pytorch

基于Pytorch的文本分类框架，支持TextCNN、Bert、Electra等。
57 stars 9 forks source link

Test #8

Open dhl961123 opened 1 year ago

dhl961123 commented 1 year ago

请问测试的时候只是将运行模式mode = 'train'改为mode = 'test'就ok吗？ 请问如何加载已训练的模型； 针对这个错误的解法可以提供一下吗？ output = outputs[0] NameError: name 'outputs' is not defined

wzzzd commented 1 year ago

1.代码默认加载 checkpoints/step_best下的模型，也就是训练过程在验证集上效果最好的模型 2.最新的代码已经修复这个问题啦

dhl961123 commented 1 year ago

好的，感谢