Ross Wightman dfa9298b4e Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl ...

* refactor 'same' convolution and add helper to use MixedConv2d when needed
* improve performance of 'same' padding for cases that can be handled statically
* add support for extra exp, pw, and dw kernel specs with grouping support to decoder/string defs for MixNet
* shuffle some args for a bit more consistency, a little less clutter overall in gen_efficientnet.py

2019-07-25 11:42:01 -07:00

..

__init__.py

Add basic image folder style dataset to read directly out of tar files, example in validate.py

2019-07-25 10:51:03 -07:00

config.py

…

constants.py

…

dataset.py

Add basic image folder style dataset to read directly out of tar files, example in validate.py

2019-07-25 10:51:03 -07:00

distributed_sampler.py

…

loader.py

Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl

2019-07-25 11:42:01 -07:00

mixup.py

…

random_erasing.py

Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print

2019-06-29 10:03:13 -07:00

tf_preprocessing.py

Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl

2019-07-25 11:42:01 -07:00

transforms.py

Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print

2019-06-29 10:03:13 -07:00