import torch
import torch.nn as nn
import torchvision as tv
from torch.utils import data
from torchvision import transforms as T
import os
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
import tensorboardX
from utils import show_confMat, validate
device = torch.device("cuda"if torch.cuda.is_available()else"cpu")
transform = T.Compose([T.Resize(224),
T.ToTensor()])# 获取训练数据集
data_train = tv.datasets.MNIST(
root='./data/mnist/',
train=True,
download=True