f2006b2437
Cleanup qkv_bias cat in beit model so it can be traced
3 years ago
1076a65df1
Minor post FX merge cleanup
3 years ago
32c9937dec
Merge branch 'fx-feature-extract-new' of https://github.com/alexander-soare/pytorch-image-models into alexander-soare-fx-feature-extract-new
3 years ago
78b36bf46c
Places365 doesn't exist in some still used torchvision version
3 years ago
65d827c7a6
rename notrace registration and standardize trace_utils imports
3 years ago
9b2daf2a35
Add ResNeXt-50 weights 81.1 top-1 @ 224, 82 @ 288 with A1 'high aug' recipe
3 years ago
0a1ff9744a
Merge branch 'martinsbruveris-add-vit-b8'
3 years ago
9b5d6dc7e2
Merge branch 'add-vit-b8' of https://github.com/martinsbruveris/pytorch-image-models into martinsbruveris-add-vit-b8
3 years ago
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
3 years ago
5220711d87
Added B/8 models to ViT.
3 years ago
0262a0e8e1
fx ready for review
3 years ago
871cef4198
version 0.5.1
3 years ago
4f338556d8
Fixes and improvements for metrics, tfds parser, loader / transform handling
...
* add back ability to create transform with loader
* change 'samples' -> 'examples' for tfds wrapper to match tfds naming
* add support for specifying feature names for input and target in tfds wrapper
* add class_to_idx for image classification datasets in tfds wrapper
* add accumulate_type to avg meters and metrics to allow float32 or float64 accumulation control with lower prec data
* minor cleanup, log output rate prev and avg
3 years ago
d2994016e9
Add try/except guards
3 years ago
b25ff96768
wip - pre-rebase
3 years ago
e051dce354
Make all models FX traceable
3 years ago
cf4561ca72
Add FX based FeatureGraphNet capability
3 years ago
0149ec30d7
wip - attempting to rebase
3 years ago
02c3a75a45
wip - make it possible to use fx graph in train and eval mode
3 years ago
a6c24b936b
Tests to enforce all models FX traceable
3 years ago
bc3d4eb403
wip -rebase
3 years ago
ab3ac3f25b
Add FX based FeatureGraphNet capability
3 years ago
d9b0b3d60f
device arg wasn't removed from PrefetcherCuda instantiation of RE
3 years ago
80ca078aed
Fix a few bugs and formatting/naming issues
...
* Pass optimizer resume flag through to checkpoint / updater restore. Related to #961 but not clear how relates to crash.
* Rename monitor step args, cleanup handling of step_end_idx vs num_steps for consistent log output in either case
* Resume from proper epoch (ie next epoch relative to checkpoint)
3 years ago
65419f60cc
Merge pull request #964 from rwightman/more_datasets
...
Dataset additions
3 years ago
406c486ba2
Merge remote-tracking branch 'origin/more_datasets' into bits_and_tpu
3 years ago
07693f81b0
Validation fix since we don't have multi-GPU DataParallel support yet
3 years ago
9ec3210c2d
More TFDS parser cleanup, support improved TFDS even_split impl (on tfds-nightly only currently).
3 years ago
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 )
3 years ago
59a3409182
Update README.md
3 years ago
ddc29da974
Add ResNet101 and ResNet152 weights from higher aug RSB recipes. 81.93 and 82.82 top-1 at 224x224.
3 years ago
b328e56f49
Update eca_halonext26ts weights to a better set
3 years ago
2ddef942b9
Better fix for #954 that doesn't break torchscript, pull torch._assert into timm namespace when it exists
3 years ago
4f0f9cb348
Fix #954 by bringing traceable _assert into timm to allow compat w/ PyTorch < 1.8
3 years ago
a41de1f666
Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738 .
3 years ago
ed41d32637
Add repr to auto_augment and random_erasing impl
3 years ago
135a48d024
Fix sam result again for imagenetv2
3 years ago
aaff2d82d0
Add new 50ts attn models to benchmark/meta csv files
3 years ago
1e17863b7b
Fixed botne*t26 model results, add some 50ts self-attn variants
3 years ago
ae72d009fa
Add weights for lambda_resnet50ts, halo2botnet50ts, lamhalobotnet50ts, updated halonet50ts
3 years ago
a45186a6e8
Merge remote-tracking branch 'origin/master' into bits_and_tpu
3 years ago
13178ba73a
Add benchmark and metadata csv files
3 years ago
b745d30a3e
Fix formatting of last commit
3 years ago
3478f1d7f1
Traceability fix for vit models for some experiments
3 years ago
f658a72e72
Cleanup re-use of Dropout modules in Mlp modules after some twitter feedback :p
3 years ago
71f00bfe9e
Don't run profile if model is torchscripted
3 years ago
7da1b0b61c
Merge pull request #933 from t-vi/unbind
...
use .unbind instead of explicitly listing the indices
3 years ago
5882e62ada
Add activation count to fvcore based profiling in benchmark.py
3 years ago
51f488b7f5
Update results.csv files with latest weights
3 years ago
f805ba86d9
use .unbind instead of explicitly listing the indices
3 years ago
Ross Wightman