pytorch学习(一):对cifar10数据分类
import torch
import torchvision
from torch import nn
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
from torchvision.transforms import transforms
import time
# 下载训练集和验证集
train_data = torchvision.datasets.CIFAR10(
原创
2022-04-02 11:12:09 ·
1285 阅读 ·
0 评论