5 Commits

Author	SHA1	Message	Date
Ross Wightman	a5a2ad2e48	Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size ... * add test that model supports forward_head(x, pre_logits=True) * add head_hidden_size attr to all models and set differently from num_features attr when head has hidden layers * test forward_features() feat dim == model.num_features and pre_logits feat dim == self.head_hidden_size * more consistency in reset_classifier signature, add typing * asserts in some heads where pooling cannot be disabled Fix #2194	2024-06-07 13:53:00 -07:00
Ross Wightman	3c0283f9ef	Fix reparameterize for NextViT. Fix #2187	2024-05-27 14:48:58 -07:00
Ross Wightman	c838c4233f	Add typing to reset_classifier() on other models	2024-05-12 11:12:00 -07:00
Ross Wightman	935950cc11	Fix F.sdpa attn drop prob	2024-02-10 20:14:47 -08:00
Ross Wightman	0737cf231d	Add Next-ViT	2024-02-10 17:05:16 -08:00