在Torch中保存lstm语言模型
2016-3-17 22:58:46
收藏:0
阅读:50
评论:0
我正在使用lstm语言模型,代码来自https://github.com/wojzaremba/lstm/blob/master/main.lua
我想要在训练结束后保存模型,以备后续使用。我在训练结束的位置加入了以下代码:
torch.save(params.model_file, model)
看起来这行代码非常成功地保存了模型。但是,当我尝试加载该模型并进行测试时,得到的困惑度非常大。仅仅为了测试,我运行了一个小实例进行训练,结果得到的测试集困惑度为134,然后保存了模型。然后我加载了保存的模型,并在相同的测试集上使用完全相同的测试方法(函数run_test),但是我得到了一个巨大的困惑度,达到了71675.134(即使使用随机权重的结果也比这个要低得多!)。我尝试保存和加载只有权重,将它们转换为float()之后再保存,或者将它们保存为cudaTensors,但是我得到的结果都是一样的。
下面是在保存了整个模型后加载并测试的代码;我只修改了原始main.lua中的主函数:
local function main()
g_init_gpu(arg)
print('从文件中加载模型' .. params.model_file)
model=torch.load(params.model_file)
state_test = {data=transfer_data(ptb.testdataset(params.batch_size))}
reset_state(state_test)
run_test()
end
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- 如何将两个不同的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中获取用户配置主目录的跨平台方法