caffe ssd test阶段显示loss

SSD训练有一个很奇怪的“疏忽”,就是训练过程没有默认显示TEST loss,以至于我们并不清楚模型是否过拟合

苦苦搜寻这个解决方案,在https://github.com/weiliu89/caffe/issues/449 里有人提到了这个问题,而且给出了解决方案:
in the python script,when you create the test net, after CreateMultiBoxHead, add these lines

name = "test_loss"
mbox_layers.append(net.label)
net[name] = L.MultiBoxLoss(*mbox_layers, multibox_loss_param=multibox_loss_param, loss_param=loss_param, include=dict(phase=caffe_pb2.Phase.Value('TEST')), propagate_down=[True, True, False, False])
name="silence"
net[name] = L.Silence(net['test_loss'],ntop=0)

also, add this to the solver_param:
'test_compute_loss':True,

但是我自己尝试在python文件添加这几行的时候
hehe.jpg



运行却报了很奇怪的错
hehehe.jpg


我交换过训练和验证数据集,还是会出这样的问题,一直解决不了,请问有人解决过这个问题么,谢谢
已邀请:

要回复问题请先登录注册