You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Ross Wightman
477a78ed81
Fix optimizer factory regressin for optimizers like sgd/momentum that don't have an eps arg
|
4 years ago |
| .. |
|
data
|
More models in sotabench, more control over sotabench run, dataset filename extraction consistency
|
4 years ago |
|
loss
|
asymmetric_loss
|
4 years ago |
|
models
|
Improve test crop for ViT models. Small now 77.85, added base weights at 79.35 top-1.
|
4 years ago |
|
optim
|
Fix optimizer factory regressin for optimizers like sgd/momentum that don't have an eps arg
|
4 years ago |
|
scheduler
|
Merge branch 'logger' into features. Change 'logger' to '_logger'.
|
4 years ago |
|
utils
|
Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.
|
4 years ago |
|
__init__.py
|
Monster commit, activation refactor, VoVNet, norm_act improvements, more
|
4 years ago |
|
version.py
|
Add missing leaky_relu layer factory defn, update Apex/Native loss scaler interfaces to support unscaled grad clipping. Bump ver to 0.2.2 for pending release.
|
4 years ago |
