使用 `apply` 方法在 `CudaTensor` 上出现异常
2016-3-13 4:29:45
收藏:0
阅读:63
评论:1
这个问题只发生在尝试使用 torch 的 CUDA 能力而不是依赖 CPU 之后。
我正在尝试初始化一个卷积神经网络的权重 CudaTensor。该函数如下所示:
function fill_0normal(t,sigma)
t:apply(function() return torch.normal(0,sigma) end)
end
并且是通过以下方式调用的:
fill_0normal(m.weight, sigma)
其中 m 是卷积模块,m.weight 是一个 CudaTensor,sigma 是一个浮点数。
我得到的异常如下所示:
/hpc/sw/torch7-2016.02.09/bin/luajit: invalid arguments: number number
expected arguments: *CudaTensor* [float] [float]
stack traceback:
[C]: at 0x2aaaaf63e040
[C]: in function 'func'
/hpc/sw/torch7-2016.02.09/share/lua/5.1/torch/FFI.lua:117: in function 'apply'
/hpc/sw/torch7-2016.02.09/share/lua/5.1/cutorch/Tensor.lua:3: in function 'apply'
setup_model.lua:4: in function 'fill_0normal'
setup_model.lua:16: in function 'init_conv'
setup_model.lua:43: in function 'init_module'
setup_model.lua:90: in function 'initializeNetRandomly'
assignment3-cifar10.lua:49: in main chunk
[C]: in function 'dofile'
.../torch7-2016.02.09/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk
[C]: at 0x00406010
有什么想法是什么可能会导致它?
我还尝试将 torch.normal 返回的值转换为一个 1 元素的 CudaTensor,但这没有帮助。
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- 如何将两个不同的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中获取用户配置主目录的跨平台方法
你可以直接在张量上调用
:normal。function fill_0normal(t,sigma) do t:normal(0, sigma) end