在 CNN 之上的 LSTM
2016-6-10 12:33:47
收藏:0
阅读:67
评论:1
我已经在 Torch 中实现了以下 LSTM 模型,取自以下链接: https://github.com/wojzaremba/lstm/blob/master/main.lua
我有一个关于下面代码的问题:
local function create_network()
local x = nn.Identity()()
local y = nn.Identity()()
local prev_s = nn.Identity()()
local i = {[0] = LookupTable(params.vocab_size,
params.rnn_size)(x)}
local next_s = {}
local split = {prev_s:split(2 * params.layers)}
for layer_idx = 1, params.layers do
local prev_c = split[2 * layer_idx - 1]
local prev_h = split[2 * layer_idx]
local dropped = nn.Dropout(params.dropout)(i[layer_idx - 1])
local next_c, next_h = lstm(dropped, prev_c, prev_h)
table.insert(next_s, next_c)
table.insert(next_s, next_h)
i[layer_idx] = next_h
end
local h2y = nn.Linear(params.rnn_size, params.vocab_size)
local dropped = nn.Dropout(params.dropout)(i[params.layers])
local pred = nn.LogSoftMax()(h2y(dropped))
local err = nn.ClassNLLCriterion()({pred, y})
local module = nn.gModule({x, y, prev_s},
{err, nn.Identity()(next_s)})
module:getParameters():uniform(-params.init_weight, params.init_weight)
return transfer_data(module)
end
在处理 ptb 数据库的嵌入部分中,代码在处理时使用了 LookupTable 层,现在我想知道如何使用 LookupTable 来定义不同类型数据的其他嵌入。特别是,输入是 RGB 图像,嵌入将是其中一种 CNN 模型,例如没有全连接层的 AlexNet。
对于我来说,它看起来太模糊了。 有没有更好的设计来达到这个目的? 如何创建 CNN 之上的 LSTM 模型?
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1:1导入真实世界的地形?
- 求解,lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教:如何用lua实现游戏内调用数据库函数实现账号密码注册?
- Lua实现网站后台开发
- LUA错误显式返回,社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接,并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下,有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值(不考虑键的顺序)
- 有个lua简单的项目,外包,有意者加微信 liuheng600456详谈,最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码?
- addEventListener 返回 nil Lua
- Lua中获取用户配置主目录的跨平台方法
Torch nn.LookupTable简单来说就是在其权重张量上执行 Tensor 索引。在你指定的代码中,它也用于学习单词向量,因为它包含在 nngraph 模型中。如果你有一个预训练模型,你可以将其权重设置为 LookupTable,但这次你不应该在 nngraph 中包含它。权重张量的维度应该是 nIndex(例如,你有多少个不同的图像)x nOutput(例如,LSTM 隐藏层大小 - 代码中的 rnn_size)。或者,你也可以直接指定输入张量而不使用 LookupTable。