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 5f81d4de23 Move DeiT to own file, vit getting crowded. Working towards fixing #1029, make pooling interface for transformers and mlp closer to convnets. Still working through some details...		2 years ago
..
__init__.py	Basic forward pass test for all registered models	4 years ago
test_layers.py	Fix #661, move hardswish out of default args for LeViT. Enable native torch support for hardswish, hardsigmoid, mish if present.	3 years ago
test_models.py	Move DeiT to own file, vit getting crowded. Working towards fixing #1029, make pooling interface for transformers and mlp closer to convnets. Still working through some details...	2 years ago
test_optim.py	Merge branch 'master' into attn_update	3 years ago
test_utils.py	Fix ordering of tests	3 years ago