Ross Wightman c1a84ecb22 dataset not passed through PrefetchLoader for inference script. Fix #10 ...

* also, make top5 configurable for lower class count cases

2019-06-11 21:59:56 -07:00

..

__init__.py

Mixup and prefetcher improvements

2019-05-18 22:17:51 -07:00

config.py

…

constants.py

…

dataset.py

Ported Tensorflow pretrained EfficientNet weights and some model cleanup

2019-05-30 17:55:35 -07:00

distributed_sampler.py

…

loader.py

dataset not passed through PrefetchLoader for inference script. Fix #10

2019-06-11 21:59:56 -07:00

mixup.py

Mixup and prefetcher improvements

2019-05-18 22:17:51 -07:00

random_erasing.py

Change args for RandomErasing so only one required for pixel/color mode

2019-05-18 12:29:30 -07:00

tf_preprocessing.py

Ported Tensorflow pretrained EfficientNet weights and some model cleanup

2019-05-30 17:55:35 -07:00

transforms.py

Mixup and prefetcher improvements

2019-05-18 22:17:51 -07:00