search

luckyhzt / unicolor

This is the implementation of paper: "UniColor: A Unified Framework for Multi-Modal Colorization with Transformer"
63 stars 9 forks source link

I'm having difficulties when trying to train UniColor on my own datasets #6

Open TTXhaC opened 1 year ago

TTXhaC commented 1 year ago

Dear author, First of all,Thanks a lot for this project！ And I‘m very interested in UniColor，But when trying to train UniColor on my own datasets，I encounter some difficulties. Now I have done the data preprocessing，and run through train_vqgan.py and train_tran.py, but I didn't find the file that calls the functions defined by them，I wonder how to aquire the checkpoints like the pre-trained model offered in this project. Will there be a unified file like 'train.py'?I can‘t wait to see the instructions of train to gain a deeper understanding of this project,could u release the instructions of train or 'train.py' earlier? Best, Xiaoha

xy-cong commented 1 year ago

Same! hope the instructions of train coming soonn!!!