Commit Graph

63 Commits (3b9032ea481eb61bcc4afd04d84f5dd83d1029bb)

Author SHA1 Message Date
Ross Wightman 3cdaf5ed56 Add `mmax` config key to auto_augment for increasing upper bound of RandAugment magnitude beyond 10. Make AugMix uniform sampling default not override config setting. 3 years ago
Ross Wightman d53e91218e Fix tf.data options setting for newer TF versions 4 years ago
Norman Mu 79640fcc1f Enable uniform augmentation magnitude sampling and set AugMix default 4 years ago
Ross Wightman a5310a3451 Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 4 years ago
Ross Wightman f42f1df26c Improve evenness of per-worker split for validation set with TFDS 4 years ago
Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs. 4 years ago
Ross Wightman 2db2d87ff7 Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only. 4 years ago
Ross Wightman 68a4144882 Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg. 4 years ago
Ross Wightman 22748f1a2d Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py 4 years ago
Ross Wightman 5d4c3d0af3 Add enhanced ParserImageInTar that can read images from tars within tars, folders with multiple tars, etc. Additional comment cleanup. 4 years ago
Ross Wightman 855d6cc217 More dataset work including factories and a tensorflow datasets (TFDS) wrapper 4 years ago
Ross Wightman 59ec7e6a53 Merge branch 'master' into imagenet21k_datasets_more 4 years ago
Ross Wightman e7a9ddf982
Merge pull request from kecsap/links 4 years ago
Csaba Kertesz 7cae7e7035 Follow links during dataset scanning 4 years ago
Ross Wightman 0167f749d3 Remove some old __future__ imports 4 years ago
Ross Wightman e35e9760a6 More work on dataset / parser split and imagenet21k (tar) support 4 years ago
Ross Wightman de6046e213 Initial commit for dataset / parser reorg to support additional datasets / types 4 years ago
Tymoteusz Wiśniewski de15b43865 Fix a bug with accuracy retrieving from RealLabels 4 years ago
Ross Wightman e8ca45854c More models in sotabench, more control over sotabench run, dataset filename extraction consistency 4 years ago
Ross Wightman 47a7b3b5b1 More flexible mixup mode, add 'half' mode. 4 years ago
Ross Wightman 8c9814e3f5 Final cleanup of mixup/cutmix. Element/batch modes working with both collate (prefetcher active) and without prefetcher. 4 years ago
Ross Wightman cd23f55397 Fix mixed prec issues with new mixup code 4 years ago
Ross Wightman f471c17c9d More cutmix/mixup overhaul, ready to kick-off some trials. 4 years ago
Ross Wightman 92f2d0d65d Merge branch 'master' into cutmix. Fixup a few issues. 4 years ago
Ross Wightman e62758cf4f More documentation updates, fix a typo 5 years ago
Ross Wightman fa28067704 Add more augmentation arguments, including a no_aug disable flag. Fix 5 years ago
Ross Wightman c53ec33ae0 Add synset/label indices for results generation. Add 'valid labels' to validation script to support imagenet-a/r label subsets properly. 5 years ago
Ross Wightman 7995295968 Merge branch 'logger' into features. Change 'logger' to '_logger'. 5 years ago
Ross Wightman 6c17d57a2c Fix some attributions, add copyrights to some file docstrings 5 years ago
Ross Wightman d23a2697d0 Working on feature extraction, interfaces refined, a number of models working, some in progress. 5 years ago
Antoine Broyelle 78fa0772cc Leverage python hierachical logger 5 years ago
AFLALO, Jonathan Isaac a7f570c9b7 added MultiEpochsDataLoader 5 years ago
Ross Wightman 64fe37d008 Remove old mean/std helper, rely fully on cmd line or default_cfg now. Fixes 5 years ago
Ross Wightman 670c61b28f Some cutmix/mixup cleanup/fixes 5 years ago
Ross Wightman b3cb5f3275 Working on CutMix impl as per , integrating with Mixup, currently experimenting... 5 years ago
Ross Wightman d66819d1f3 Indentation mistake. Fixes 5 years ago
Ross Wightman 1daa303744 Add support to Dataset for class id mapping file, clean up a bit of old logic. Add results file arg for validation and update script. 5 years ago
Ross Wightman d9a6a9d0af
Merge pull request from rwightman/augmix-jsd 5 years ago
Ross Wightman 3eb4a96eda Update AugMix, JSD, etc comments and references 5 years ago
Ross Wightman 7547119891 Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together 5 years ago
Ross Wightman 2e955cfd0c Update RandomErasing with some improved arg names, tweak to aspect range 5 years ago
Ross Wightman 3cc0f91e23 Fix augmix variable name scope overlap, default non-blended mode 5 years ago
Ross Wightman 4666cc9aed Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script 5 years ago
Ross Wightman 3afc2a4dc0 Some cleanup/improvements to AugMix impl: 5 years ago
Ross Wightman 232ab7fb12 Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl 5 years ago
Ross Wightman 0161de0127 Switch RandoErasing back to on GPU normal sampling 5 years ago
Ross Wightman 31453b039e Update Auto/RandAugment comments, README, more. 5 years ago
Ross Wightman 4243f076f1 Adding RandAugment to AutoAugment impl, some tweaks to AA included 5 years ago
Ross Wightman 4002c0d4ce Fix AutoAugment abs translate calc 5 years ago
Ross Wightman c06274e5a2 Add note on random selection of magnitude value 5 years ago