2021年2月2日
学习笔记14：模型保存
摘要： 保存训练过程中使得测试集上准确率最高的参数 import copy best_model_wts = copy.deepcopy(model.state_dict()) best_acc = 0 train_loss = [] train_acc = [] test_loss = [] test_a 阅读全文
posted @ 2021-02-02 14:42 pbc的成长之路 阅读(245) 评论(0) 推荐(0)
学习笔记13：微调模型
摘要： resnet预训练模型 resnet模型与之前笔记中的vgg模型不同，需要我们直接覆盖掉最后的全连接层 先看一下resnet模型的结构： 我们需要先将所有的参数都设置成requires_grad = False 然后再重新定义fc层，并覆盖掉原来的。 重新定义的fc层的requires_grad默认 阅读全文
posted @ 2021-02-02 10:54 pbc的成长之路 阅读(403) 评论(0) 推荐(0)
学习笔记12：图像数据增强及学习速率衰减
摘要： 数据增强 常用数据增强方法： transforms.RandomCrop # 随机位置裁剪 transforms.CenterCrop # 中心位置裁剪 transforms.RandomHorizontalFlip(p = 1) # 随机水平翻转 transforms.RandomVertical 阅读全文
posted @ 2021-02-02 08:58 pbc的成长之路 阅读(254) 评论(0) 推荐(0)