слой Python с Caffe застрял после создания слоя - PullRequest
Новая
       21

слой Python с Caffe застрял после создания слоя

2 голосов
/ 19 марта 2019

Я использую linreg.prototxt, который использует EuclideanLossLayer, предоставленный в папке examples / pycaffe, со следующей командой: 

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

Однако после этого код не выполняется и застревает.

В чем может быть причина?Спасибо.

...