yun-liu / RCF

Richer Convolutional Features for Edge Detection
Other
758 stars 260 forks source link

Layer conv1_1 has unknown engine. #12

Closed wanghuok closed 7 years ago

wanghuok commented 7 years ago

hi. thanks for your great job. I got this error when i run RCF-multiscale.ipynb.

I0911 19:19:47.419523 17941 layer_factory.hpp:77] Creating layer input I0911 19:19:47.419531 17941 net.cpp:101] Creating Layer input I0911 19:19:47.419535 17941 net.cpp:409] input -> data I0911 19:19:47.428836 17941 net.cpp:151] Setting up input I0911 19:19:47.428869 17941 net.cpp:158] Top shape: 1 3 224 224 (150528) I0911 19:19:47.428874 17941 net.cpp:166] Memory required for data: 602112 I0911 19:19:47.428879 17941 layer_factory.hpp:77] Creating layer data_input_0_split I0911 19:19:47.428887 17941 net.cpp:101] Creating Layer data_input_0_split I0911 19:19:47.428890 17941 net.cpp:435] data_input_0_split <- data I0911 19:19:47.428896 17941 net.cpp:409] data_input_0_split -> data_input_0_split_0 I0911 19:19:47.428903 17941 net.cpp:409] data_input_0_split -> data_input_0_split_1 I0911 19:19:47.428908 17941 net.cpp:409] data_input_0_split -> data_input_0_split_2 I0911 19:19:47.428912 17941 net.cpp:409] data_input_0_split -> data_input_0_split_3 I0911 19:19:47.428917 17941 net.cpp:409] data_input_0_split -> data_input_0_split_4 I0911 19:19:47.428961 17941 net.cpp:151] Setting up data_input_0_split I0911 19:19:47.428966 17941 net.cpp:158] Top shape: 1 3 224 224 (150528) I0911 19:19:47.428969 17941 net.cpp:158] Top shape: 1 3 224 224 (150528) I0911 19:19:47.428972 17941 net.cpp:158] Top shape: 1 3 224 224 (150528) I0911 19:19:47.428977 17941 net.cpp:158] Top shape: 1 3 224 224 (150528) I0911 19:19:47.428979 17941 net.cpp:158] Top shape: 1 3 224 224 (150528) I0911 19:19:47.428982 17941 net.cpp:166] Memory required for data: 3612672 I0911 19:19:47.428983 17941 layer_factory.hpp:77] Creating layer conv1_1 F0911 19:19:47.428990 17941 layer_factory.cpp:69] Layer conv1_1 has unknown engine. Check failure stack trace: Aborted (core dumped)

what's wrong with it? thanks a lot.

yun-liu commented 7 years ago

The reason is that you did not compile with CUDNN. You can simply solve this problem by removing the engine: CUDNN in prototxt.

wanghuok commented 7 years ago

it works. thanks.