jasneik的博客

import mathimport matplotlib.pyplot as pltimport numpy as npimport torchimport torch.nn as nnfrom torch.optim import *from torchvision import modelsclass Net(nn.Module): def __init__(self): super(Net,self).__init__() self.fc = n