Training code for GeM baseline

Where is a training code to train GeM baseline?

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.

thanks my issue has been fixed.