Commit Graph

77 Commits (d66b4c23554fe1f33fbb8873fffa0ecd1faceff1)

Author SHA1 Message Date
Ross Wightman 656757d26b Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones. 3 years ago
Ross Wightman 02ae11e526 Leaving repeat aug sampler indices as tensor thrashes worker shared process memory 3 years ago
Hyeongchan Kim a0b2657497
Use `torch.repeat_interleave()` to generate repeated indices faster () 3 years ago
Ross Wightman 78b36bf46c Places365 doesn't exist in some still used torchvision version 3 years ago
Ross Wightman cfa414cad2 Matching two bits_and_tpu changes for TFDs wrapper 3 years ago
Ross Wightman 9ec3210c2d More TFDS parser cleanup, support improved TFDS even_split impl (on tfds-nightly only currently). 3 years ago
Ross Wightman ba65dfe2c6 Dataset work 3 years ago
Ross Wightman a41de1f666 Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix . 3 years ago
Ross Wightman ed41d32637 Add repr to auto_augment and random_erasing impl 3 years ago
masafumi 047a5ec05f Fix bugs that Mixup does not work device=cpu 3 years ago
Ross Wightman 80075b0b8a Add worker_seeding arg to allow selecting old vs updated data loader worker seed for (old) experiment repeatability 3 years ago
Ross Wightman f8a63a3b71 Add worker_init_fn to loader for numpy seed per worker 3 years ago
Ross Wightman fb94350896 Update training script and loader factory to allow use of scheduler updates, repeat augment, and bce loss 3 years ago
Ross Wightman f262137ff2 Add RepeatAugSampler as per DeiT RASampler impl, showing promise for current (distributed) training experiments. 3 years ago
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 4 years ago
Ross Wightman fa28067704 Add more augmentation arguments, including a no_aug disable flag. Fix 4 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