使用训练好的模型做分类任务,两百多个文件,因为还有其他任务要处理,所以每个文件都要加载一次模型,但是越往后加载的越慢,有什么好的解决办法吗
微信里点“发现”,扫一下
二维码便可将本文分享至朋友圈。
1,每次加载模型前清理图 2,训练时保存权重,预测时加载权重