Deep-Exemplar-based-Colorization icon indicating copy to clipboard operation
Deep-Exemplar-based-Colorization copied to clipboard

Need to change caffe source code for correct running

Open MSKTHU opened this issue 5 years ago • 0 comments

For some reasons, I cannot use docker. Recently, I use this https://github.com/ncianeo/Deep-Exemplar-based-Colorization/tree/linux-docker-cv-caffe-build repo and have successfully run it under linux 18.04 without docker. However, I have to change the source code of your supplied caffe. Otherwise, there will be segmentation fault, for both deepanalogy and similarity_net. The reason of segfault for deepanalogy is the force_backward param in its prototxt. And the reason for similarity_net is the following code in caffe/src/caffe/net.cpp, funtion AppendParam().

    has_params_lr_.push_back(param_spec->has_lr_mult());
    has_params_decay_.push_back(param_spec->has_decay_mult());
    params_lr_.push_back(param_spec->lr_mult());
    params_weight_decay_.push_back(param_spec->decay_mult());

Do you know the reason? Or could you please show your env?

MSKTHU avatar Jan 26 '21 10:01 MSKTHU