solo-learn icon indicating copy to clipboard operation
solo-learn copied to clipboard

gradient checkpointing with timm

Open DonkeyShot21 opened this issue 2 years ago • 1 comments

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

DonkeyShot21 avatar Nov 14 '22 17:11 DonkeyShot21

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

vturrisi avatar Jan 29 '23 13:01 vturrisi