11-深度学习模型架构-基础版-cpu

    import torch
    from torch import nn
    from torch import optim
    from torch.nn import functional as F
    import torchvision
    
    from matplotlib import pyplot
    
    batch_size = 512
    # 1-数据加载
    transform = torchvision.transforms.Compose(
    [torchvision.transforms.ToTensor(), torchvision.transforms.Normalize((0.1307,), (0.3081,))])
    train_datasets = torchvision.datasets.MNIST('mnist', train=True, download=True, transform=transform)
    train_loader = torch.utils.data.DataLoader(train_datasets, batch_size=batch_size, shuffle=True)
    test_datasets = t