Ross Wightman 9c297ec67d Cleanup Apex vs native AMP scaler state save/load. Cleanup CheckpointSaver a bit.

2020-09-02 15:12:59 -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

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

2020-09-02 15:12:59 -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