摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14361926.html 保存训练过程中使得测试集上准确率最高的参数 import copy best_model_wts = copy.deepcopy(model.state_dict()) best_acc = 阅读全文
posted @ 2024-06-04 09:29 有何m不可 阅读(44) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14360807.html resnet预训练模型 resnet模型与之前笔记中的vgg模型不同,需要我们直接覆盖掉最后的全连接层先看一下resnet模型的结构: 我们需要先将所有的参数都设置成requires_grad 阅读全文
posted @ 2024-06-04 09:28 有何m不可 阅读(72) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14360231.html 数据增强 常用数据增强方法: transforms.RandomCrop # 随机位置裁剪 transforms.CenterCrop # 中心位置裁剪 transforms.RandomHo 阅读全文
posted @ 2024-06-04 09:27 有何m不可 阅读(33) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14348536.html 什么是预训练网络 预训练模型就是之前用较大的数据集训练出来的模型,这个模型通过微调,在另外类似的数据集上训练。一般预训练模型规模比较大,训练起来占用大量的内存资源。 微调预训练网络 我们采用v 阅读全文
posted @ 2024-06-04 09:26 有何m不可 阅读(66) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14348160.html 相关包导入 import torch import numpy as np import pandas as pd import matplotlib.pyplot as plt import 阅读全文
posted @ 2024-06-04 09:16 有何m不可 阅读(64) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14345342.html 相关包导入 import torch import pandas as pd import numpy as np import matplotlib.pyplot as plt from t 阅读全文
posted @ 2024-06-04 09:13 有何m不可 阅读(42) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14344935.html 相关包导入 import torch import pandas as pd import numpy as np import matplotlib.pyplot as plt from t 阅读全文
posted @ 2024-06-04 09:12 有何m不可 阅读(81) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14335456.html 相关包 import torch import pandas as pd import numpy as np import matplotlib.pyplot as plt from tor 阅读全文
posted @ 2024-06-04 09:10 有何m不可 阅读(26) 评论(0) 推荐(0)
摘要: 转自:https://www.cnblogs.com/miraclepbc/p/14335416.html 导入相关包 import torch import pandas as pd import numpy as np import matplotlib.pyplot as plt from t 阅读全文
posted @ 2024-06-04 09:09 有何m不可 阅读(23) 评论(0) 推荐(0)
摘要: TensorDataset 转自:https://www.cnblogs.com/miraclepbc/p/14333299.html 导入相关包 from torch.utils.data import TensorDataset 特征与标签合并 HRdataset = TensorDataset 阅读全文
posted @ 2024-06-04 09:07 有何m不可 阅读(111) 评论(0) 推荐(0)