首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
emdedding专题
Pytorch实现多层LSTM模型,并增加emdedding、Dropout、权重共享等优化
简述 本文是 Pytorch封装简单RNN模型,进行中文训练及文本预测 一文的延申,主要做以下改动: 1.将nn.RNN替换为nn.LSTM,并设置多层LSTM: 既然使用pytorch了,自然不需要手动实现多层,注意nn.RNN和nn.LSTM 在实例化时均有参数num_layers来指定层数,本文设置num_layers=2; 2.新增emdedding层,替换掉原来的nn.funct
阅读更多...