search

richzhang / colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
http://richzhang.github.io/colorization/
BSD 2-Clause "Simplified" License
3.34k stars 923 forks source link

the last layer type 'SoftmaxCrossEntropyLoss' is an unknown type #55

Closed beautifulSun closed 5 years ago

beautifulSun commented 5 years ago

Sorry to intrude. When i training this model,i have got an error called "Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer type: SoftmaxCrossEntropyLoss". I get Caffe by running ./train/fetch_caffe.sh.I have found the softmax_cross_entropy_loss_layer.cpp and softmax_cross_entropy_loss_layer.cu in caffe folder.So,How to use these two files in my training process? Thanks.

beautifulSun commented 5 years ago

oh,it looks like be solved by myself.