Problem setting up a layer: Check failed ... layer inputs must have the same count.
springyworks opened this issue · comments
running the given train.sh , I get the next failure, what can I do?
(not using GPU )
.......
I0730 17:25:47.946137 5825 net.cpp:120] Setting up loss
F0730 17:25:47.949908 5825 sigmoid_cross_entropy_loss_layer.cpp:26] Check failed: bottom[0]->count() == bottom[1]->count() (2097152 vs. 8388608) SIGMOID_CROSS_ENTROPY_LOSS layer inputs must have the same count.
*** Check failure stack trace: ***
@ 0x7ff780704daa (unknown)
@ 0x7ff780704ce4 (unknown)
@ 0x7ff7807046e6 (unknown)
@ 0x7ff780707687 (unknown)
@ 0x7ff780aeeb7d caffe::SigmoidCrossEntropyLossLayer<>::Reshape()
@ 0x7ff780b3e772 caffe::Net<>::Init()
@ 0x7ff780b404d2 caffe::Net<>::Net()
@ 0x7ff780b027d0 caffe::Solver<>::InitTrainNet()
@ 0x7ff780b03773 caffe::Solver<>::Init()
@ 0x7ff780b03946 caffe::Solver<>::Solver()
@ 0x40c920 caffe::GetSolver<>()
@ 0x406571 train()
@ 0x404ab1 main
@ 0x7ff77fc16ec5 (unknown)
@ 0x40505d (unknown)
@ (nil) (unknown)
./train.sh: line 1: 5825 Aborted (core dumped) caffe train -solver numbers_solver.prototxt
solution: in file numbers_net.autoencoder.prototxt change line#238 to reflect the size of the spectral images