InstColorization
InstColorization copied to clipboard
Fix AttributeError while loading weights and poor colorization results if cuda is disabled
arg with: --gpu_ids -1
Tested on colab.
Fix: poor colorizing results by using pre-trained weights if cuda is disabled.
This patch just resolve model loading, "CPU gray" stands there, not change any more.
This patch just resolve model loading, "CPU gray" stands there, not change any more.
I've tested on both my laptop and Google CoLab by using pretrained weight and can't reproduce this problem. Could you tell me about weights and options you use on testing?