facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Other
6.93k stars 1.17k forks source link

Monitor training of custom dataset #165

Open DanielShalam opened 1 year ago

DanielShalam commented 1 year ago

Hello, thanks for you great work !

I'm trying to train a custom dataset using your method, Problem is that I'm not sure how to monitor the performance when pretraining ? In other works, knn between epochs commonly used, but its not clear if its the right way to go, considering the difference of you method.

Any suggestions?