Code

ViT pooling refactor#2209

Code

Merged

rwightman merged 4 commits intomainhuggingface/pytorch-image-models:mainfrom fcossio-vit-maxpoolhuggingface/pytorch-image-models:fcossio-vit-maxpoolCopy head branch name to clipboardJun 17, 2024

+73-34

Commits

Commits on Jun 14, 2024

• Feature: add option global_pool='max' to VisionTransformer

  Show description for 9567cf6
  fcossio

  authored
  9567cf6

  Copy full SHA for 9567cf6

Commits on Jun 15, 2024

• Refactor vit pooling to add more reduction options, separately callable

  rwightman

  committed
  71101eb

  Copy full SHA for 71101eb

Commits on Jun 16, 2024

• Some missed reset_classifier() type annotations

  rwightman

  committed
  b1a6f4a

  Copy full SHA for b1a6f4a

• Add numpy<2.0 to requirements until tests are sorted out for pytorch 2.3 vs older

  rwightman

  committed
  6254dfa

  Copy full SHA for 6254dfa