首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何计算pycaffe的测试/验证损失

如何计算pycaffe的测试/验证损失
EN

Stack Overflow用户
提问于 2016-07-21 18:13:06
回答 1查看 630关注 0票数 0

我试图在python中计算我自己的训练循环中的测试损失。调用solver.test_nets[0].forward()似乎更新了score blob,但没有更新loss blob。知道怎么更新吗?

我使用以下解决程序配置:

代码语言:javascript
运行
复制
net: "/tmp/tmp8ikb9sg2/train.prototxt"
test_net: "/tmp/tmp8ikb9sg2/test.prototxt"
test_iter: 1
test_interval: 2147483647
base_lr: 0.1
lr_policy: "fixed"
test_initialization: false

除了文件顶部的阶段定义外,train和test.prototxt完全相同:

代码语言:javascript
运行
复制
name: "pycaffenet"
state {
  phase: TRAIN  # set TEST in test.prototxt
}
...
layer {
  name: "loss"
  type: "SoftmaxWithLoss"
  bottom: "score"
  bottom: "output"
  top: "loss"
}
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2016-08-01 10:27:55

其实是另一个问题,我的想法。loss blob正在更新,但由于solver.test_nets[0]的权重没有变化,所以它保持不变。看起来它们不会自动与solver.net共享。这可以通过简单地调用:

代码语言:javascript
运行
复制
solver.test_nets[0].share_with(solver.net)
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/38511503

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档