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 c2cd1a332e Improve torch amp support and add channels_last support for train/validate scripts		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	Fix MobileNetV3 crash with global_pool='', output consistent with other models but not equivalent due to efficient head.	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