InstColorization icon indicating copy to clipboard operation
InstColorization copied to clipboard

Published 20 hours ago • verified publisher icon ericsujw

Fix AttributeError while loading weights and poor colorization results if cuda is disabled

Open thelastlin opened this issue 4 years ago • 3 comments

arg with: --gpu_ids -1

Tested on colab.

thelastlin avatar Aug 05 '20 14:08 thelastlin

Fix: poor colorizing results by using pre-trained weights if cuda is disabled.

thelastlin avatar Aug 06 '20 10:08 thelastlin

This patch just resolve model loading, "CPU gray" stands there, not change any more.

delldu avatar Nov 25 '20 15:11 delldu

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?

thelastlin avatar Nov 30 '20 06:11 thelastlin