import torchvision.models
import torch
import time
from torch import nn
from torchvision.models import resnet18, ResNet18_Weights
from torch.utils.data import DataLoader
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
data_transform=torchvision.transforms.ToTensor()
train_set=torchvision.datasets.CIFAR10(root="./dataset",train=True,download=False,transform=data_transform
使用resnet18预训练模型做CIFAR10
最新推荐文章于 2024-09-08 21:07:51 发布