摘要:
import torch import torchvision import matplotlib.pyplot as plt import torch.nn as nn import torch.nn.functional as F from torch.utils.data import Dat 阅读全文
摘要:
使用pytorch。 一、定义模型结构 model.py import torch.nn as nn import torch.nn.functional as F """ 定义一个类,这个类继承于nn.Module,实现两个方法:初始化函数和正向传播 实例化这个类之后,将参数传入这个类中,进行正向 阅读全文