Train on custom dataset

[zhaotf16]

Hi, Zhiyuan. Thanks for your contribution! But I don't know how to train a model on my own dataset. If possible, can you tell me how to tune the program? Thank you!

[ZhiyuanDang]

Hi,
first, you should create a pre-train config file same as files in configs/pretext. Then, you train a pre-train model using this file.

After that, you follow the usages section in Readme.md.

At the same time, you should create your model and dataset file the same as files in models/ and data/.