搜索内容

最新文章
  • 两百行的图像分类,用CIFAR10训练ResNet18
    Python

    两百行的图像分类,用CIFAR10训练ResNet18

    已阅: 6 水一下这个月的稿子,网上肯定很多,但写都写了就发出来。 # train.py from torch.utils.data import DataLoader from torch.autograd import Variable import torchvision.models as models import torchvision.transforms as T import torch.nn.functional as F import matplotlib.pyplot as plt import numpy as np import torchvision import argparse import random import torch import time import s……
    MWHLS 2024/07/11