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

Open

thelastlin wants to merge 2 commits intoericsujw:masterfrom

thelastlin:master

Commits on Aug 5, 2020

Fix AttributeError while loading weights if cuda is disabled (--gpu_ids -1)
thelastlin
committed

Commits on Aug 6, 2020

Fix: poor results due to failed to load weights.
thelastlin
committed