实战PyTorch(二++):宝可梦自定义数据集之迁移学习

Load data Build model Train and Test Transfer Learning 思想:若是本身建立resnet18网络,也就是从0开始训练,只能达到85的精度,由于任务的图片数量少,而网络表达能力强,可能出现过拟合的状况,若是用训练好的reset18,有着公有知识,这样训练不是从0开始,而是从公有知识开始,能获得更好的状态。简单来讲,也就是在A上训练好分类器后,转移到
相关文章
相关标签/搜索