linghu8812的博客

文章目录1. 需要用到的库2. 模型定义3. 测试 基于上一篇文章https://blog.youkuaiyun.com/linghu8812/article/details/100044971，这次介绍一下网络模型的定义。 1. 需要用到的库 import torch import torch.nn as nn from torchvision.models import resnet50 基于ResNet50定义网络模型，在此基础上进行微调。 2. 模型定义 class CatVSDogNet(nn.Module

1.需要用到的库 设置学习率和模型 import math import matplotlib.pyplot as plt import torch.optim as optim from torchvision.models import resnet18 lr_rate = 0.1 model = resnet18(num_classes=10) 2.LambdaLR实现cosine...