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 f31933cb37 Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.		4 years ago
..
data	More models in sotabench, more control over sotabench run, dataset filename extraction consistency	4 years ago
loss	Fix some attributions, add copyrights to some file docstrings	4 years ago
models	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	4 years ago
optim	Add model based wd skip support. Improve cross version compat of optimizer factory. Fix #247	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	5 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