21 Commits (cf324ea38f7a537992b3f61c22f73ec927c899d1)

Author	SHA1	Message	Date
Ross Wightman	c061d5e401	Allow using class_map functionality w/ IterableDataset (TFDS/WDS) to remap class labels	1 year ago
Ross Wightman	e9dccc918c	Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc	2 years ago
Ross Wightman	f67a7ee8bd	Set num_workers in Iterable WDS/TFDS datasets early so sample estimate is correct	2 years ago
Ross Wightman	b8c8550841	Data improvements. Improve train support for in_chans != 3. Add wds dataset support from bits_and_tpu branch w/ fixes and tweaks. TFDS tweaks.	2 years ago
Ross Wightman	87939e6fab	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	2 years ago
Ross Wightman	656757d26b	Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.	2 years ago
Ross Wightman	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
Ross Wightman	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
Ross Wightman	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
Ross Wightman	e35e9760a6	More work on dataset / parser split and imagenet21k (tar) support	3 years ago
Ross Wightman	de6046e213	Initial commit for dataset / parser reorg to support additional datasets / types	3 years ago
Ross Wightman	e8ca45854c	More models in sotabench, more control over sotabench run, dataset filename extraction consistency	4 years ago
Ross Wightman	92f2d0d65d	Merge branch 'master' into cutmix. Fixup a few issues.	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.	4 years ago
Ross Wightman	6c17d57a2c	Fix some attributions, add copyrights to some file docstrings	4 years ago
Ross Wightman	b3cb5f3275	Working on CutMix impl as per #8, integrating with Mixup, currently experimenting...	4 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.	4 years ago
Ross Wightman	7547119891	Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together	4 years ago
Ross Wightman	232ab7fb12	Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl	4 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	aa4354f466	Big re-org, working towards making pip/module as 'timm'	5 years ago