ab16a358bb
Add log and continue handler for WDS errors, fix args.num_gpu for validation script fallback
2 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
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
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
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
e0b3a3fab3
Make test-pooling flag for validate.py opt in
3 years ago
25d52ea71d
Merge remote-tracking branch 'origin/fixes_bce_regnet' into bits_and_tpu
3 years ago
5d6983c462
Batch validate a list of files if model is a text file with model per line
3 years ago
f2e14685a8
Add force-cpu flag for train/validate, fix CPU fallback for device init, remove old force cpu flag for EMA model weights
3 years ago
40457e5691
Transforms, augmentation work for bits, add RandomErasing support for XLA (pushing into transforms), revamp of transform/preproc config, etc ongoing...
3 years ago
91ab0b6ce5
Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go...
3 years ago
74d2829341
Merge branch 'master' into bits_and_tpu
3 years ago
aa92d7b1c5
Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes.
3 years ago
12d9a6d4d2
First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working
3 years ago
a04427d8ce
Add _in22k to bulk validate filter
3 years ago
a5310a3451
Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update
3 years ago
d584e7f617
Support for huggingface hub via create_model and default_cfgs.
...
* improve consistency of model creation helper fns
* add comments to some of the model helpers
* support passing external default_cfgs so they can be sourced from hub
3 years ago
2db2d87ff7
Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.
3 years ago
d8e69206be
Merge pull request #419 from rwightman/byob_vgg_models
...
More models, GPU-Efficient Nets, RepVGG, classic VGG, and flexible Byob backbone.
3 years ago
0356e773f5
Default to native PyTorch AMP instead of APEX amp. Too many APEX issues cropping up lately.
3 years ago
b4e216e377
Fix a few small things.
3 years ago
5114c214fc
Change the Python interpreter to Python 3.x in the scripts
3 years ago
2a8c4dc63b
Add validation script update for using test_input_size in model default_cfgs
3 years ago
38d8f67570
Fix potential issue with change to num_classes arg in train/validate.py defaulting to None (rely on model def / default_cfg)
3 years ago
855d6cc217
More dataset work including factories and a tensorflow datasets (TFDS) wrapper
...
* Add parser/dataset factory methods for more flexible dataset & parser creation
* Add dataset parser that wraps TFDS image classification datasets
* Tweak num_classes handling bug for 21k models
* Add initial deit models so they can be benchmarked in next csv results runs
3 years ago
59ec7e6a53
Merge branch 'master' into imagenet21k_datasets_more
3 years ago
e42b140ade
Add --input-size option to scripts to specify full input dimensions from command-line
3 years ago
ce69de70d3
Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models
3 years ago
de6046e213
Initial commit for dataset / parser reorg to support additional datasets / types
3 years ago
8ffdc5910a
test_time_pool would be set to a non-False value even if test-time pooling is not available
4 years ago
e8ca45854c
More models in sotabench, more control over sotabench run, dataset filename extraction consistency
4 years ago
751b0bba98
Add global_pool (--gp) arg changes to allow passing 'fast' easily for train/validate to avoid channels_last issue with AdaptiveAvgPool
4 years ago
9c297ec67d
Cleanup Apex vs native AMP scaler state save/load. Cleanup CheckpointSaver a bit.
4 years ago
c2cd1a332e
Improve torch amp support and add channels_last support for train/validate scripts
4 years ago
ad150e7018
Update results csv file rank/diff script and small validate script tweak for batch validation
4 years ago
ecdeb470f2
Cleanup real labels in validate.py, no need to track original, will do that via exteran script
4 years ago
c53ec33ae0
Add synset/label indices for results generation. Add 'valid labels' to validation script to support imagenet-a/r label subsets properly.
4 years ago
7995295968
Merge branch 'logger' into features. Change 'logger' to '_logger'.
4 years ago
1998bd3180
Merge branch 'feature/AB/logger' of https://github.com/antoinebrl/pytorch-image-models into logger
4 years ago
6c17d57a2c
Fix some attributions, add copyrights to some file docstrings
4 years ago
4e61c6a12d
Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript
4 years ago
d23a2697d0
Working on feature extraction, interfaces refined, a number of models working, some in progress.
4 years ago
d3ee3de96a
Update validation script first batch prime and clear cuda cache between multi-model runs
4 years ago
78fa0772cc
Leverage python hierachical logger
...
with this update one can tune the kind of logs generated by timm but
training and inference traces are unchanged
4 years ago
88129b2569
Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.
4 years ago
eb7653614f
Monster commit, activation refactor, VoVNet, norm_act improvements, more
...
* refactor activations into basic PyTorch, jit scripted, and memory efficient custom auto
* implement hard-mish, better grad for hard-swish
* add initial VovNet V1/V2 impl, fix #151
* VovNet and DenseNet first models to use NormAct layers (support BatchNormAct2d, EvoNorm, InplaceIABN)
* Wrap IABN for any models that use it
* make more models torchscript compatible (DPN, PNasNet, Res2Net, SelecSLS) and add tests
4 years ago
02a30411ad
Replace fp16 with amp support for validate.py script
4 years ago
13cf68850b
Remove poorly named metrics from torch imagenet example origins. Use top1/top5 in csv output for consistency with existing validation results files, acc elsewhere. Fixes #111
4 years ago
53c47479c4
Batch validation batch size adjustment, tweak L2 crop pct
4 years ago
Ross Wightman
Ross Wightman
Csaba Kertesz
Matthijs Hollemans
Antoine Broyelle