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 f944242cb0 Fix #262, num_classes arg mixup. Make vision_transformers a bit closer to other models wrt get/reset classfier/forward_features. Fix torchscript for ViT.		4 years ago
..
__init__.py	…
adafactor.py	…
adahessian.py	…
adamp.py	…
adamw.py	…
lookahead.py	…
nadam.py	…
novograd.py	…
nvnovograd.py	…
optim_factory.py	Fix #262, num_classes arg mixup. Make vision_transformers a bit closer to other models wrt get/reset classfier/forward_features. Fix torchscript for ViT.	4 years ago
radam.py	…
rmsprop_tf.py	…
sgdp.py	…