mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
1,493 Commits 141 Branches 65 Tags
Commit Graph

1493 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Ross Wightman
734b2244fe Add RegNetZ-D8 (83.5 @ 256, 84 @ 320) and RegNetZ-E8 (84.5 @ 256, 85 @ 320) weights. Update names of existing RegZ models to include group size. 2021-11-20 15:52:04 -08:00
Ross Wightman
05092e2fbe Add more models to FX filter 2021-11-20 15:51:48 -08:00
Ross Wightman
93cc08fdc5 Make evonorm variables 1d to match other PyTorch norm layers, will break weight compat for any existing use (likely minimal, easy to fix). 2021-11-20 15:50:51 -08:00
Ross Wightman
af607b75cc Prep a set of ResNetV2 models with GroupNorm, EvoNormB0, EvoNormS0 for BN free model experiments on TPU and IPU 2021-11-19 17:37:00 -08:00
Ross Wightman
3819bef93e Add FX test exclusion since it uses more ram and barfs on GitHub actions. Will take a few iterations to include needed models :( 2021-11-19 17:35:41 -08:00
Ross Wightman
c976a410d9 Add ResNet-50 w/ GN (resnet50_gn) and SEBotNet-33-TS (sebotnet33ts_256) model defs and weights. Update halonet50ts weights w/ slightly better variant in1k val, more robust to test sets. 2021-11-19 14:24:43 -08:00
Ross Wightman
9b3519545d Attempt to reduce memory footprint of FX tests for GitHub actions runs 2021-11-19 14:24:12 -08:00
Ross Wightman
bdd3dff0ca beit_large models killing GitHub actions test, filter out 2021-11-19 08:39:48 -08:00
Ross Wightman
9d6aad44f8 Update tests to run Python 3.9, PyTorch 1.10, torchvision 0.11.1 2021-11-18 21:38:04 -08:00
Ross Wightman
ee40b582bb Merge branch 'alexander-soare-fx-feature-extract-new' 2021-11-18 21:26:56 -08:00
Ross Wightman
f2006b2437 Cleanup qkv_bias cat in beit model so it can be traced 2021-11-18 21:25:00 -08:00
Ross Wightman
1076a65df1 Minor post FX merge cleanup 2021-11-18 19:47:07 -08:00
Ross Wightman
32c9937dec Merge branch 'fx-feature-extract-new' of https://github.com/alexander-soare/pytorch-image-models into alexander-soare-fx-feature-extract-new 2021-11-18 16:31:29 -08:00
Ross Wightman
78b36bf46c Places365 doesn't exist in some still used torchvision version 2021-11-18 14:59:51 -08:00
Alexander Soare
65d827c7a6 rename notrace registration and standardize trace_utils imports 2021-11-15 21:03:21 +00:00
Ross Wightman
9b2daf2a35 Add ResNeXt-50 weights 81.1 top-1 @ 224, 82 @ 288 with A1 'high aug' recipe 2021-11-14 13:17:27 -08:00
Ross Wightman
0a1ff9744a Merge branch 'martinsbruveris-add-vit-b8' 2021-11-14 13:07:16 -08:00
Ross Wightman
9b5d6dc7e2 Merge branch 'add-vit-b8' of https://github.com/martinsbruveris/pytorch-image-models into martinsbruveris-add-vit-b8 2021-11-14 12:54:28 -08:00
Ross Wightman
cfa414cad2 Matching two bits_and_tpu changes for TFDs wrapper 
* change 'samples' -> 'examples' for tfds wrapper to match tfds naming
* add class_to_idx for image classification datasets in tfds wrapper
 2021-11-14 12:52:19 -08:00
Martins Bruveris
5220711d87 Added B/8 models to ViT. 2021-11-14 11:01:48 +00:00
Alexander Soare
0262a0e8e1 fx ready for review 2021-11-13 00:06:33 +00:00
Alexander Soare
d2994016e9 Add try/except guards 2021-11-12 21:16:53 +00:00
Alexander Soare
b25ff96768 wip - pre-rebase 2021-11-12 20:45:05 +00:00
Alexander Soare
e051dce354 Make all models FX traceable 2021-11-12 20:45:05 +00:00
Alexander Soare
cf4561ca72 Add FX based FeatureGraphNet capability 2021-11-12 20:45:05 +00:00
Alexander Soare
0149ec30d7 wip - attempting to rebase 2021-11-12 20:45:05 +00:00
Alexander Soare
02c3a75a45 wip - make it possible to use fx graph in train and eval mode 2021-11-12 20:45:05 +00:00
Alexander Soare
a6c24b936b Tests to enforce all models FX traceable 2021-11-12 20:45:05 +00:00
Alexander Soare
bc3d4eb403 wip -rebase 2021-11-12 20:45:05 +00:00
Alexander Soare
ab3ac3f25b Add FX based FeatureGraphNet capability 2021-11-12 20:45:05 +00:00
Ross Wightman
65419f60cc
Merge pull request #964 from rwightman/more_datasets 
Dataset additions
 2021-11-11 13:34:05 -08:00
Ross Wightman
9ec3210c2d More TFDS parser cleanup, support improved TFDS even_split impl (on tfds-nightly only currently). 2021-11-10 15:52:09 -08:00
Ross Wightman
ba65dfe2c6 Dataset work 
* support some torchvision datasets
* improvements to TFDS wrapper for subsplit handling (fix #942), shuffle seed
* add class-map support to train (fix #957)
 2021-11-09 22:34:15 -08:00
Ross Wightman
ddc29da974 Add ResNet101 and ResNet152 weights from higher aug RSB recipes. 81.93 and 82.82 top-1 at 224x224. 2021-11-02 17:59:16 -07:00
Ross Wightman
b328e56f49 Update eca_halonext26ts weights to a better set 2021-11-02 16:52:53 -07:00
Ross Wightman
2ddef942b9 Better fix for #954 that doesn't break torchscript, pull torch._assert into timm namespace when it exists 2021-11-02 11:22:33 -07:00
Ross Wightman
4f0f9cb348 Fix #954 by bringing traceable _assert into timm to allow compat w/ PyTorch < 1.8 2021-11-02 09:21:40 -07:00
Ross Wightman
a41de1f666 Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738. 2021-10-28 17:35:01 -07:00
Ross Wightman
ed41d32637 Add repr to auto_augment and random_erasing impl 2021-10-28 17:33:36 -07:00
Ross Wightman
135a48d024 Fix sam result again for imagenetv2 2021-10-28 14:46:45 -07:00
Ross Wightman
aaff2d82d0 Add new 50ts attn models to benchmark/meta csv files 2021-10-28 14:32:47 -07:00
Ross Wightman
1e17863b7b Fixed botne*t26 model results, add some 50ts self-attn variants 2021-10-28 13:55:24 -07:00
Ross Wightman
ae72d009fa Add weights for lambda_resnet50ts, halo2botnet50ts, lamhalobotnet50ts, updated halonet50ts 2021-10-27 22:08:54 -07:00
Ross Wightman
13178ba73a Add benchmark and metadata csv files 2021-10-26 10:50:46 -07:00
Ross Wightman
b745d30a3e Fix formatting of last commit 2021-10-25 15:15:14 -07:00
Ross Wightman
3478f1d7f1 Traceability fix for vit models for some experiments 2021-10-25 15:13:08 -07:00
Ross Wightman
f658a72e72 Cleanup re-use of Dropout modules in Mlp modules after some twitter feedback :p 2021-10-25 00:40:59 -07:00
Ross Wightman
71f00bfe9e Don't run profile if model is torchscripted 2021-10-24 22:41:20 -07:00
Ross Wightman
7da1b0b61c
Merge pull request #933 from t-vi/unbind 
use .unbind instead of explicitly listing the indices
 2021-10-24 22:33:18 -07:00
Ross Wightman
5882e62ada Add activation count to fvcore based profiling in benchmark.py 2021-10-24 15:30:38 -07:00