search

vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
MIT License
1.39k stars 181 forks source link

ConvNeXt-V2 implementations #322

Closed mbenami closed 1 year ago

mbenami commented 1 year ago

Hi is there any plan to add support for ConvNeXt-V2 training? I saw there is already implementation on timm

Thanks a lot for this fantastic work!

vturrisi commented 1 year ago

Hey! Thanks for the kind words. Adding new methods is always something we plan on doing, however we base it on the time that it would take for implementing it and how interested in the method is the community. I think convnext V2 should be pretty straightforward to add. I'll take a look at the original code and start to draft an implementation in the next couple of weeks.

Edit: I think that what timm has is the pretrained model.

vturrisi commented 1 year ago

Closing this for now. Would be happy to assist, but it's not a priority for me right now.