DIGITS: Train Caffe Model task failed with error code -9.

Hi I was trying out DIGITS to train a caffe model. I have built caffe using just CPU and running digits-devserver with --config flag. The weird thing was I wasnt asked the options to use Digits with just the cpu even with the flag. After the job is initialised I get the following message.

norm2 needs backward computation.
relu2 needs backward computation.
conv2 needs backward computation.
pool1 needs backward computation.
norm1 needs backward computation.
relu1 needs backward computation.
conv1 needs backward computation.
label_val-data_1_split does not need backward computation.
val-data does not need backward computation.
This network produces output accuracy
This network produces output loss
Network initialization done.
Solver scaffolding done.
Starting Optimization
Solving
Learning Rate Policy: step
Iteration 0, Testing net (#0)
Ignoring source layer train-data
Test net output #0: accuracy = 0.5
Test net output #1: loss = 0.69282 (* 1 = 0.69282 loss)

Please Help.

About this issue

  • Original URL
  • State: open
  • Created 7 years ago
  • Comments: 17 (8 by maintainers)

Most upvoted comments

Do you have CUDA installed?