Ross Wightman 751b0bba98 Add global_pool (--gp) arg changes to allow passing 'fast' easily for train/validate to avoid channels_last issue with AdaptiveAvgPool

2020-09-02 16:13:47 -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 global_pool (--gp) arg changes to allow passing 'fast' easily for train/validate to avoid channels_last issue with AdaptiveAvgPool

2020-09-02 16:13:47 -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

Cleanup Apex vs native AMP scaler state save/load. Cleanup CheckpointSaver a bit.

2020-09-02 15:12:59 -07:00

version.py

Bump version to 0.2.1 and update README

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