You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 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
..
parsers	Fixes and improvements for metrics, tfds parser, loader / transform handling	3 years ago
__init__.py	Fixes and improvements for metrics, tfds parser, loader / transform handling	3 years ago
auto_augment.py	Merge remote-tracking branch 'origin/more_datasets' into bits_and_tpu	3 years ago
collate.py	Transforms, augmentation work for bits, add RandomErasing support for XLA (pushing into transforms), revamp of transform/preproc config, etc ongoing...	4 years ago
config.py	Transforms, augmentation work for bits, add RandomErasing support for XLA (pushing into transforms), revamp of transform/preproc config, etc ongoing...	4 years ago
constants.py	…
dataset.py	Dataset work	3 years ago
dataset_factory.py	Dataset work	3 years ago
distributed_sampler.py	Add RepeatAugSampler as per DeiT RASampler impl, showing promise for current (distributed) training experiments.	3 years ago
fetcher.py	Remove print, arg order	4 years ago
loader.py	Fixes and improvements for metrics, tfds parser, loader / transform handling	3 years ago
mixup.py	Fixes and improvements for metrics, tfds parser, loader / transform handling	3 years ago
prefetcher_cuda.py	device arg wasn't removed from PrefetcherCuda instantiation of RE	3 years ago
random_erasing.py	Add repr to auto_augment and random_erasing impl	3 years ago
real_labels.py	Fix a bug with accuracy retrieving from RealLabels	4 years ago
tf_preprocessing.py	First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working	4 years ago
transforms.py	Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738.	3 years ago
transforms_factory.py	Fixes and improvements for metrics, tfds parser, loader / transform handling	3 years ago