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
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
5 years ago
..
__init__.py
Add basic image folder style dataset to read directly out of tar files, example in validate.py
5 years ago
config.py
A number of tweaks to arguments, epoch handling, config
5 years ago
constants.py
Big re-org, working towards making pip/module as 'timm'
5 years ago
dataset.py
Add basic image folder style dataset to read directly out of tar files, example in validate.py
5 years ago
distributed_sampler.py
Big re-org, working towards making pip/module as 'timm'
5 years ago
loader.py
Add MixNet ( https://arxiv.org/abs/1907.09595 ) with pretrained weights converted from Tensorflow impl
5 years ago
mixup.py
Big re-org, working towards making pip/module as 'timm'
5 years ago
random_erasing.py
Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print
5 years ago
tf_preprocessing.py
Add MixNet ( https://arxiv.org/abs/1907.09595 ) with pretrained weights converted from Tensorflow impl
5 years ago
transforms.py
Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print
5 years ago