自然语言处理 基于预训练模型的方法 代码解读 [ section 4.1.6, page 77 ]
自定义神经网络模型的例子:一个简单的多层感知器模型MLP
文件名为mlp.py
# Defined in Section 4.1.6
# Page 77
# 自定义神经网络模型的第一个例子(The first example of custom neural network model)
# 导入torch和torch.nn
# pytorch将常用的神经网络模型封装到了torch.nn包内,从而可以方便灵活地加以调用
# Sigmoid, Softmax, ReLU, tanh等各种激活函数包含.
原创
2022-04-29 16:15:24 ·
1385 阅读 ·
0 评论