Ross Wightman 80c9d9cc72 Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast'

2020-09-02 09:11:48 -07:00

..

data

Final cleanup of mixup/cutmix. Element/batch modes working with both collate (prefetcher active) and without prefetcher.

2020-08-12 17:01:32 -07:00

loss

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

models

Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast'

2020-09-02 09:11:48 -07:00

optim

Merge branch 'logger' into features. Change 'logger' to '_logger'.

2020-07-27 18:00:46 -07:00

scheduler

Merge branch 'logger' into features. Change 'logger' to '_logger'.

2020-07-27 18:00:46 -07:00

__init__.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

utils.py

Improve torch amp support and add channels_last support for train/validate scripts

2020-08-31 17:58:16 -07:00

version.py

Bump version to 0.2.1 and update README

2020-08-12 18:05:07 -07:00