Ross Wightman 460eba7f24 Work around casting issue with combination of native torch AMP and torchscript for Linear layers

2020-11-30 13:30:51 -08:00

..

data

…

loss

asymmetric_loss

2020-10-16 17:12:28 +03:00

models

Work around casting issue with combination of native torch AMP and torchscript for Linear layers

2020-11-30 13:30:51 -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