代码: import torch.nn as nn
import numpy as npfrom matplotlib import pyplot as plt
import time
#from utils import get_accur,load_data,trainimport torchvision
from torchvision import transforms
from torch.utils.data import DataLoader
import torch…