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
4666cc9aed
Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script
|
6 years ago |
| .. |
|
__init__.py
|
Adding RandAugment to AutoAugment impl, some tweaks to AA included
|
6 years ago |
|
auto_augment.py
|
Update Auto/RandAugment comments, README, more.
|
6 years ago |
|
config.py
|
A number of tweaks to arguments, epoch handling, config
|
6 years ago |
|
constants.py
|
Big re-org, working towards making pip/module as 'timm'
|
6 years ago |
|
dataset.py
|
Add basic image folder style dataset to read directly out of tar files, example in validate.py
|
6 years ago |
|
distributed_sampler.py
|
Big re-org, working towards making pip/module as 'timm'
|
6 years ago |
|
loader.py
|
Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script
|
6 years ago |
|
mixup.py
|
Big re-org, working towards making pip/module as 'timm'
|
6 years ago |
|
random_erasing.py
|
Switch RandoErasing back to on GPU normal sampling
|
6 years ago |
|
tf_preprocessing.py
|
Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl
|
6 years ago |
|
transforms.py
|
Update Auto/RandAugment comments, README, more.
|
6 years ago |
