pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard
verified publisher icon huggingface

Support gradient checkpointing in `forward_intermediates()`

Open brianhou0208 opened this issue 6 months ago • 1 comments

Summary:

  • Renamed mask to attn_mask in NaFlex ViT forward_intermediates()
  • Removed unused imports
  • Support gradient checkpointing in forward_intermediates()", resolve #2435

brianhou0208 avatar May 28 '25 21:05 brianhou0208

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

HuggingFaceDocBuilderDev avatar May 29 '25 14:05 HuggingFaceDocBuilderDev