5 Commits

Author	SHA1	Message	Date
Ross Wightman	5d599a6a10	RepViT weights on HF hub	2023-08-25 10:39:02 -07:00
Ross Wightman	c692715388	Some RepVit tweaks ... * add head dropout to RepVit as all models have that arg * default train to non-distilled head output via distilled_training flag (set_distilled_training) so fine-tune works by default w/o distillation script * camel case naming tweaks to match other models	2023-08-09 12:41:12 -07:00
alec.tu	bb2b6b5f09	fix num_classes not found	2023-08-07 15:16:03 +08:00
Ross Wightman	b71d60cdb7	Two small fixes, num_classes in base class, add model tag	2023-07-26 13:18:49 -07:00
jameslahm	3318e7614d	Add RepViT models	2023-07-21 14:56:53 +08:00