hmmlillian / Neural-Color-Transfer

The source code of "single-reference color transfer" in "Progressive Color Transfer with Dense Semantic Correspondences".
https://arxiv.org/abs/1710.00756
72 stars 8 forks source link

Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR #7

Open juneleung opened 1 year ago

juneleung commented 1 year ago

Hello, thanks for your awesome work! I run with run.bat but some errors happened, hope to get your guide, thanks!

··· I1014 21:52:04.348300 21188 layer_factory.hpp:77] Creating layer input I1014 21:52:04.348300 21188 net.cpp:100] Creating Layer input I1014 21:52:04.348300 21188 net.cpp:418] input -> data I1014 21:52:04.349303 21188 net.cpp:150] Setting up input I1014 21:52:04.350303 21188 net.cpp:157] Top shape: 1 3 224 224 (150528) I1014 21:52:04.350303 21188 net.cpp:165] Memory required for data: 602112 I1014 21:52:04.351305 21188 layer_factory.hpp:77] Creating layer conv1_1 I1014 21:52:04.351305 21188 net.cpp:100] Creating Layer conv1_1 I1014 21:52:04.351305 21188 net.cpp:444] conv1_1 <- data I1014 21:52:04.352304 21188 net.cpp:418] conv1_1 -> conv1_1 F1014 21:52:04.627554 21188 cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR Check failure stack trace: ···

hmmlillian commented 1 year ago

Please try this solution.

juneleung commented 1 year ago

I tried to open TDR and set the delay to 10, but still does not work if I run with run.bat, same note with CUDNN. if I run exe directly, will get a message can't find vgg,

Check failed: fd != -1 (-1 vs. -1) File not found: \vgg19\VGG_ILSVRC_19_layers_deploy.prototxt
*** Check failure stack trace: ***

my folder is like this: image is it the correction setting?

zyyue99 commented 1 year ago

Hello, thanks for your awesome work! I run with run.bat but some errors happened, hope to get your guide, thanks!

··· I1014 21:52:04.348300 21188 layer_factory.hpp:77] Creating layer input I1014 21:52:04.348300 21188 net.cpp:100] Creating Layer input I1014 21:52:04.348300 21188 net.cpp:418] input -> data I1014 21:52:04.349303 21188 net.cpp:150] Setting up input I1014 21:52:04.350303 21188 net.cpp:157] Top shape: 1 3 224 224 (150528) I1014 21:52:04.350303 21188 net.cpp:165] Memory required for data: 602112 I1014 21:52:04.351305 21188 layer_factory.hpp:77] Creating layer conv1_1 I1014 21:52:04.351305 21188 net.cpp:100] Creating Layer conv1_1 I1014 21:52:04.351305 21188 net.cpp:444] conv1_1 <- data I1014 21:52:04.352304 21188 net.cpp:418] conv1_1 -> conv1_1 F1014 21:52:04.627554 21188 cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR Check failure stack trace: ···

Hi, I have the same question. Have you solved it yet, please?

hmmlillian commented 1 year ago

@zhaoyue99

Please try this solution.

hmmlillian commented 1 year ago

I tried to open TDR and set the delay to 10, but still does not work if I run with run.bat, same note with CUDNN. if I run exe directly, will get a message can't find vgg,

Check failed: fd != -1 (-1 vs. -1) File not found: \vgg19\VGG_ILSVRC_19_layers_deploy.prototxt
*** Check failure stack trace: ***

my folder is like this: image is it the correction setting?

The missing vgg file is placed under this folder: demo/model/vgg19/VGG_ILSVRC_19_layers_deploy.prototxt.

zyyue99 commented 1 year ago

@zhaoyue99

Please try this solution.

Thank you very much for your reply! I've tried, but the problem still hasn't been solved. I am worried that the command I executed was wrong. Could you please provide me with an example command? Thank you very much!

hmmlillian commented 1 year ago

@zhaoyue99

Please try this solution.

Thank you very much for your reply! I've tried, but the problem still hasn't been solved. I am worried that the command I executed was wrong. Could you please provide me with an example command? Thank you very much!

Please follow the instruction under Demo step by step and it includes the the example command like this:

neural_color_transfer.exe -m [MODEL_DIR] -i [INPUT_ROOT_DIR] -o [OUTPUT_DIR] -g [GPU_ID]
e.g., exe\neural_color_transfer.exe -m model\ -i example\ -o example\res\ -g 0
deladeladela commented 3 weeks ago

我尝试打开 TDR 并将延迟设置为 10,但如果我使用 run.bat 运行,仍然不起作用,CUDNN 也是如此。 如果我直接运行 exe,将收到一条消息“找不到 vgg”,

Check failed: fd != -1 (-1 vs. -1) File not found: \vgg19\VGG_ILSVRC_19_layers_deploy.prototxt
*** Check failure stack trace: ***

我的文件夹是这样的: 是校正设置吗? 图像

请问解决了吗?