pytorch-image-models

Author	SHA1	Message	Date
Alexander Soare	bc3d4eb403	wip -rebase	3 years ago
Ross Wightman	4d96165989	Merge branch 'master' into cleanup_xla_model_fixes	3 years ago
Ross Wightman	8880f696b6	Refactoring, cleanup, improved test coverage. * Add eca_nfnet_l2 weights, 84.7 @ 384x384 * All 'non-std' (ie transformer / mlp) models have classifier / default_cfg test added * Fix #694 reset_classifer / num_features / forward_features / num_classes=0 consistency for transformer / mlp models * Add direct loading of npz to vision transformer (pure transformer so far, hybrid to come) * Rename vit_deit* to deit_* * Remove some deprecated vit hybrid model defs * Clean up classifier flatten for conv classifiers and unusual cases (mobilenetv3/ghostnet) * Remove explicit model fns for levit conv, just pass in arg	3 years ago
Ross Wightman	2a63d0246b	Post merge cleanup	3 years ago
Mike	b87d98b238	Update convit.py Cut out the duplicates	3 years ago
Ross Wightman	bfc72f75d3	Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights.	3 years ago
Ross Wightman	b7de82e835	ConViT cleanup, fix torchscript, bit of reformatting, reuse existing layers.	3 years ago
Aman Arora	5db1eb6ba5	Add defaults	3 years ago
Aman Arora	8b1f2e8e1f	remote unused matplotlib import	3 years ago
Aman Arora	40c506ba1e	Add ConViT	3 years ago