InstColorization icon indicating copy to clipboard operation
InstColorization copied to clipboard

Published 20 hours ago • verified publisher icon ericsujw

FineTuning Model

Open steven0704 opened this issue 3 years ago • 2 comments

First of all, great job on the project! I was wondering if there were any implemented functionality to finetune this model? Basically introducing another dataset to train on top of the already trained model.

Thanks in advance.

steven0704 avatar Oct 08 '20 08:10 steven0704

Hi, I am trying to train the model on my own dataset. After training is complete, there is no colourisation happening whatsoever. Is anyone able to train this model well and would give me some insight? Such as amount of epochs, loss and dataset size? Cheers!

steven0704 avatar Oct 21 '20 04:10 steven0704

Hi Dear friend, If you use CPU test, not GPU, you will get gray result. Just set gpu_ids = "-1" in base_options.py. Do you test on CPU ? @steven0704

delldu avatar Nov 25 '20 14:11 delldu