search

ericsujw / InstColorization

MIT License
712 stars 152 forks source link

FineTuning Model #16

Open steven0704 opened 4 years ago

steven0704 commented 4 years ago

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 commented 4 years ago

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!

delldu commented 4 years ago

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