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 c1a84ecb22 dataset not passed through PrefetchLoader for inference script. Fix #10 ... * also, make top5 configurable for lower class count cases		6 years ago
..
__init__.py	Mixup and prefetcher improvements	6 years ago
config.py	Some transform/data/loader refactoring, hopefully didn't break things	6 years ago
constants.py	Some transform/data/loader refactoring, hopefully didn't break things	6 years ago
dataset.py	Ported Tensorflow pretrained EfficientNet weights and some model cleanup	6 years ago
distributed_sampler.py	Add distributed sampler that maintains order of original dataset (for validation)	6 years ago
loader.py	dataset not passed through PrefetchLoader for inference script. Fix #10	6 years ago
mixup.py	Mixup and prefetcher improvements	6 years ago
random_erasing.py	Change args for RandomErasing so only one required for pixel/color mode	6 years ago
tf_preprocessing.py	Ported Tensorflow pretrained EfficientNet weights and some model cleanup	6 years ago
transforms.py	Mixup and prefetcher improvements	6 years ago