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

gradient checkpointing with timm #310

Open DonkeyShot21 opened 1 year ago

DonkeyShot21 commented 1 year ago

looks very easy, we can just call this function: https://github.com/rwightman/pytorch-image-models/blob/8ff45e41f7a6aba4d5fdadee7dc3b7f2733df045/timm/models/resnet.py#L697

vturrisi commented 1 year ago

Bumping this up as we plan to do it next week.