The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
COMMITS
/ tests/test_models.py September 16, 2025
R
Missed 7b filter from GITHUB specific
Ross Wightman committed
R
R
Exclude 7b from unit tests
Ross Wightman committed
July 3, 2025
G
Update no_grad usage to inference_mode if possible
GuillaumeErhard committed
June 26, 2025
R
Update test, encoder_only mode for backward test
Ross Wightman committed
June 20, 2025
R
Add basic grad checkpointing tests
Ross Wightman committed
June 4, 2025
R
Add naflex vit exceptions to tests
Ross Wightman committed
June 3, 2025
R
Merge remote-tracking branch 'origin/main' into naflex
Ross Wightman committed
May 23, 2025
R
Merge remote-tracking branch 'origin/main' into naflex
Ross Wightman committed
May 17, 2025
R
Merge branch 'main' into fast_model
Ryan committed
R
add ghostnetv3
Ryan committed
May 14, 2025
R
Merge branch 'main' into fast_model
Ryan committed
May 11, 2025
R
Merge branch 'main' into tnt
Ryan committed
May 7, 2025
R
fix pit & add to test
Ryan committed
May 5, 2025
R
update FEAT_INTER_FILTERS list
Ryan committed
R
add fasternet
Ryan committed
May 6, 2025
R
Check forward_intermediates features against forward_features output
Ross Wightman committed
May 3, 2025
R
add test_models
Ryan committed
R
Fix default_cfgs
Ryan committed
April 8, 2025
R
Exclude naflex models from jit tests
Ross Wightman committed
January 21, 2025
R
Add num_classes assertion after reset_classifier
Ryan committed
December 31, 2024
R
test filter tweaks
Ross Wightman committed
December 30, 2024
R
Update tests for aimv2 filtering
Ross Wightman committed
November 26, 2024
R
Reduce tolerance on model inference 'owl' test, pillow output varies a lot, was failing locally
Ross Wightman committed
October 2, 2024
R
R
relax tolerance on inference test
Ross Wightman committed
R
Do full inference test against test vectors for test_* models
Ross Wightman committed
August 16, 2024
R
Fixing hieradet (sam2) tests
Ross Wightman committed
July 9, 2024
R
Fixing tests
Ross Wightman committed
June 28, 2024
D
Fix jit.script breaking with features_fx
Daniel Suess committed
June 7, 2024
June 5, 2024
R
Merge remote-tracking branch 'origin/Beckschen-vitamin' into fastvit_mobileclip
Ross Wightman committed
R
Add vitamin to non-std testing models
Ross Wightman committed
May 30, 2024
R
Add fastvit to forward_intermediates test
Ross Wightman committed
May 11, 2024
R
Merge remote-tracking branch 'upstream/main' into hiera
Ross Wightman committed
May 4, 2024
R
forward_intermediates() support for byob/byoanet models
Ross Wightman committed
May 3, 2024
R
More forward_intermediates() updates
Ross Wightman committed
May 2, 2024
R
Add forward_intermediates() to efficientnet / mobilenetv3 based models as an exercise.
Ross Wightman committed
April 30, 2024
R
R
Update tests for volo/cait/xcit forward_intermediates
Ross Wightman committed
April 22, 2024
R
Disable jit test for Hiera, fair bit of work to fix.
Ross Wightman committed
April 21, 2024
R
Add forward_intermediates API to Hiera for features_only=True support
Ross Wightman committed
April 11, 2024
R
Fix / improve tests for features
Ross Wightman committed
April 10, 2024
R
More forward_intermediates() / FeatureGetterNet work
Ross Wightman committed
October 18, 2023
A
device agnostic testing
Alex Cunha committed
August 30, 2023
方
tinyvit skip ci test
方曦 committed
August 19, 2023
R
Merge pull request #1919 from ChengpengChen/main
Ross Wightman committed
August 17, 2023
方
fix bug in ci for efficientvits
方曦 committed
August 16, 2023
C
Add RepGhost models and weights
Chengpeng Chen committed
May 17, 2023
方
fix unit test for samvit
方曦 committed