我使用linreg.prototxt,它使用示例/ pycaffe文件夹中提供的EuclideanLossLayer
,并带有以下命令:
caffe train --solver solver.prototxt
solver.prototxt看起来像:
train_net: "linreg.prototxt"
base_lr: 0.001
lr_policy: "step"
gamma: 0.1
stepsize: 50000
display: 20
momentum: 0.9
weight_decay: 0.0005
snapshot: 0
snapshot_prefix: "linreg"
iter_size: 2
生成以下输出
I0319 10:22:13.395200 9167 net.cpp:86] Creating Layer ipy
I0319 10:22:13.395220 9167 net.cpp:408] ipy <- y
I0319 10:22:13.395238 9167 net.cpp:382] ipy -> ipy
I0319 10:22:13.395268 9167 net.cpp:124] Setting up ipy
I0319 10:22:13.395287 9167 net.cpp:131] Top shape: 10 10 (100)
I0319 10:22:13.395314 9167 net.cpp:139] Memory required for data: 1280
I0319 10:22:13.395342 9167 layer_factory.hpp:77] Creating layer loss
但是,此后代码无法继续执行并被卡住。
可能是什么原因?谢谢。