参考视频：在pytorch中自定义dataset读取数据_哔哩哔哩_bilibili 基本代码和视频中一模一样（视频中有提供github仓库，直接下载即可，下文代码根据实际情况略有修改） 主方法 import os from multiprocessing import freeze_support import torch from torchv…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 利用已经训练好的模型，然后给它提供输入 照一张相关图片然后传入模型，查看结果是否复合预期 import torchvision fr…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 方法1： 分别找到网络模型、数据（输入与标注等）、损失函数，然后调用.cuda函数 # 创建网络模型 pan = Pan() pan…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 完整代码见我的github仓库,完整代码中额外添加了 tensorboard的相关代码 Step1：准备数据集 train_data…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 保存 vgg16 = torchvision.models.vgg16(pretrained=False) # 保存方式1,保存模型…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 核心代码 optim = torch.optim.SGD(pan.parameters(), lr=0.01) for data i…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 损失函数的用处 1、计算实际输出和目标之间的差距 2、为我们更新输出提供一定的依据（反向传播） [[003深度学习入门——神经网络的…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 我们将模仿上图的流程建立神经网络去分类cifar10数据集 卷积层示例 [[006pyTorch入门——卷积层]] self.con…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 核心代码 class Module(nn.Module): def __init__(self): super().__init__…
合集请看：pyTorch入门合集 参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click 以ReLU和sigmoid为例 核心代码 class module(nn.Module): def __init__(self): …