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

Some transform/data/loader refactoring, hopefully didn't break things

2019-05-16 22:52:17 -07:00

constants.py

Some transform/data/loader refactoring, hopefully didn't break things

2019-05-16 22:52:17 -07:00

dataset.py

Ported Tensorflow pretrained EfficientNet weights and some model cleanup

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

distributed_sampler.py

Add distributed sampler that maintains order of original dataset (for validation)

2019-04-22 17:44:53 -07:00

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