Torch,如何在批量优化期间获取损失值的张量
2017-4-6 18:15:23
收藏:0
阅读:63
评论:1
我正在使用批量优化在训练集上进行网络训练,并且我想获取包含每个训练示例损失的损失向量。
更具体地说,我正在使用大小为3x64x64的图像,并将其大小为64的批次。 因此,我的输入是大小为64x3x64x64的张量。
在训练期间,当我编写
output = net:forward(input)
loss = criterion:forward(input,target)
loss是一个数字,但我想获得一个张量(大小为64),其中包含每个图像批次中的一项,对应于该精确图像的损失值。
有没有一种方法可以做到这一点,而不需要在输入张量的第一个维度上进行循环?
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- Lua 虚拟机加密load(string.dump(function)) 后执行失败问题如何解决
- 我想创建一个 Nginx 规则,禁止访问
- 如何将两个不同的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 代码?

forward方法调用另一个方法,即updateOutput方法,它可以被覆盖。例如,在 MSECriterion() 的情况下,您可以通过注释对THNN库的调用并编写自己想要的标准来更改此方法。也就是,对所有的元素进行常规逐元素减法,然后平方(再次逐元素)并除以总数据点数量(再次逐元素); 然后将输出作为张量返回。一旦更改了此内容,您还需要使用
luarocks make rocks/[文件夹中的 scm 文件]重新编译 nn package,然后导航到 nn 文件夹。