search

ZhiyuanDang / NNM

The PyTorch official implementation of the CVPR2021 Poster Paper NNM: Nearest Neighbor Matching for Deep Clustering.
Other
59 stars 7 forks source link

Train on custom dataset #8

Open zhaotf16 opened 2 years ago

zhaotf16 commented 2 years ago

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

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/.