Commit Graph

18 Commits (a435ea132721a388f16a63fe581057090620bd99)

Author SHA1 Message Date
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
Ross Wightman b750b76f67 More AutoAugment work. Ready to roll...
5 years ago
Ross Wightman 25d2088d9e Working on auto-augment
5 years ago
Ross Wightman f37e633e9b Merge remote-tracking branch 'origin/re-exp' into opt
5 years ago
Ross Wightman daeaa113e2 Add initial sotabench attempt. Split create_transform out of create_loader. Update requirements.txt
5 years ago
Ross Wightman 66634d2200 Add support to split random erasing blocks into randomly selected number with --recount arg. Fix random selection of aspect ratios.
5 years ago
Ross Wightman 6946281fde Experimenting with random erasing changes
5 years ago
Ross Wightman dfa9298b4e Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl
5 years ago
Ross Wightman 7a92caa560 Add basic image folder style dataset to read directly out of tar files, example in validate.py
5 years ago
Ross Wightman 6cdf35e670 Add explicit half/fp16 support to loader and validation script
5 years ago
Ross Wightman 65a634626f Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print
5 years ago
Ross Wightman c6b32cbe73 A number of tweaks to arguments, epoch handling, config
5 years ago
Ross Wightman 6fc886acaf Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
5 years ago
Ross Wightman aa4354f466 Big re-org, working towards making pip/module as 'timm'
5 years ago