Training code for GeM baseline

Where is a training code to train GeM baseline? https://github.com/facebookresearch/isc2021/blob/master/baselines/GeM_baseline.py

Hi @LexiBender, welcome to the competition. The script you linked is indeed the code for the GeM (a.k.a. MultGrain) baseline model. That model uses a pretrained model (so no training in that script) and a PCA stage that is trained on the training images.

The repo for the baseline models has some walkthroughs for using the GeM/MultiGrain model. If you’re looking for more details about the pretrained MultiGrain model, that first walkthrough page (#4) includes some links to the original MultiGrain paper and its accompanying GitHub repo.

Hope that helps!

Thank you for the response, got it!

thanks for the awesome information.