【pytorch】教程——以CIFAR10为例

有关模型的定义、参数查看和训练;卷积层计算规则;模型修改的小技巧,请参考:https://blog.csdn.net/weixin_42468475/article/details/108628474?csdn_share_tail=%7B%22type%22%3A%22blog%22%2C%22rType%22%3A%22article%22%2C%22rId%22%3A%22108628474%22%2C%22source%22%3A%22weixin_42468475%22%7D

目录

  • 1 加载数据集
  • 2 定义网络
  • 3 定义损失函数和优化器
  • 4 训练网络
  • 5 在测试集评估

1 加载数据集

torchvision库里有很多数据集。我们这次用 CIFAR10 数据集,这是一个十分类的数据集。

import torch
import torchvision
<

你可能感兴趣的:(pytorch入门到精通,pytorch,深度学习,python)