Ross Wightman fd962c4b4a Native SiLU (Swish) op doesn't export to ONNX

2020-11-29 21:56:55 -08:00

..

data

…

loss

…

models

Native SiLU (Swish) op doesn't export to ONNX

2020-11-29 21:56:55 -08:00

optim

Fix issue in optim factory with sgd / eps flag. Bump version to 0.3.1

2020-10-31 18:05:30 -07:00

scheduler

…

utils

Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.

2020-11-29 16:22:19 -08:00

__init__.py

…

version.py

ver 0.3.1

2020-10-31 18:14:58 -07:00