Ross Wightman 110a7c4982 AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992

2020-09-01 16:05:32 -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

AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992

2020-09-01 16:05:32 -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