logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

模型构造与基本深度学习基础操作2020/04/01

模型构造与基本深度学习基础操作继承MODULE类来构造模型定义MLP类重载module类的__init__ 和forward函数,分别用于创建模型和进行前向计算,无需定义反向传播函数,系统会通过自动求梯度,生成backward函数import torchimport torch.nn as nnclass MLP(nn.Module):def __init__(self):...

#深度学习#pytorch
到底了