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

ericsujw / InstColorization

MIT License

712 stars 152 forks source link

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

Open thelastlin opened 4 years ago

thelastlin commented 4 years ago

arg with: --gpu_ids -1

Tested on colab.

thelastlin commented 4 years ago

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

delldu commented 4 years ago

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

thelastlin commented 4 years ago

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?