微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Fastai export/load_learner:为什么重装后又是另一个模型

如何解决Fastai export/load_learner:为什么重装后又是另一个模型

我已经训练了一个 resnet34 学习器

tfms = get_transforms(do_flip=False)
learn = cnn_learner(data,models.resnet34,metrics=[accuracy])
learn.lr_find()
learn.recorder.plot(suggestion=True)
learn.fit_one_cycle(10,min_grad_lr)
learn.recorder.min_grad_lr
learn.export(mypath+"resnet34.pkl")

然后我计算了accuracy_score、precision_score、f1_score,我得到:0.77、0.78、0.77 之后我加载了 resnet 模型:

learn = load_learner(path=mypath,file="resnet34.pkl")

之后,我计算了accuracy_score、precision_score、f1_score(和以前一样,在exaclty上使用相同的验证集),令人惊讶的是我得到了更低的值:0.58、0.61、0.59。 我有火炬 1.5.1+cpu、torchvision 0.6.1+cpu、fastai 1.0.61 是否有任何错误/错误

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。