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.
pytorch-image-models/timm
Ross Wightman 80c9d9cc72
Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast'
4 years ago
..
data Final cleanup of mixup/cutmix. Element/batch modes working with both collate (prefetcher active) and without prefetcher. 4 years ago
loss Fix some attributions, add copyrights to some file docstrings 4 years ago
models Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast' 4 years ago
optim Merge branch 'logger' into features. Change 'logger' to '_logger'. 4 years ago
scheduler Merge branch 'logger' into features. Change 'logger' to '_logger'. 4 years ago
__init__.py Monster commit, activation refactor, VoVNet, norm_act improvements, more 4 years ago
utils.py Improve torch amp support and add channels_last support for train/validate scripts 4 years ago
version.py Bump version to 0.2.1 and update README 4 years ago