深度学习——07 快速搭建神经网络添加损失函数和优化器

最新推荐文章于 2024-07-13 00:52:38 发布

Miles强

最新推荐文章于 2024-07-13 00:52:38 发布

阅读量446

点赞数

分类专栏：深度学习文章标签：深度学习神经网络 pytorch

本文链接：https://blog.youkuaiyun.com/weixin_48501651/article/details/124792967

版权

交叉熵损失函数

loss = nn.CrossEntropyLoss()

优化器

optim = torch.optim.SGD(MyNet.parameters(),lr=0.01)
代码：

import torchvision
from torch.nn import Conv2d, MaxPool2d, Flatten, Linear, Sequential
from torch.utils.data import DataLoader
import torch.nn as nn
import torch

dataset_transform = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])
test_data = torchvision.datasets.CIFAR10(root="./test10_dataset", train