Ross Wightman 9fee316752 Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84

2020-02-24 15:11:26 -08:00

..

data

Indentation mistake. Fixes #81

2020-02-04 22:56:00 -08:00

loss

Update AugMix, JSD, etc comments and references

2020-01-11 12:02:05 -08:00

models

Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84

2020-02-24 15:11:26 -08:00

optim

Add Nvidia's NovogGrad impl from Jasper (cleaner/faster than current) and Apex Fused optimizers

2019-08-29 15:21:38 -07:00

scheduler

Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.

2020-02-22 16:23:15 -08:00

__init__.py

…

utils.py

Add map_location='cpu' to ModelEma resume, should improve #72

2020-02-12 13:23:56 -08:00

version.py

Bump version for PyPi update, fix few out of date README items/mistakes, add README updates for TF EfficientNet-B8 (RandAugment)

2020-02-03 11:44:17 -08:00