12 Commits

Author	SHA1	Message	Date
Ross Wightman	0d124ffd4f	Update README. Fine-grained layer-wise lr decay working for tiny_vit and both efficientvits. Minor fixes.	2023-09-01 15:05:29 -07:00
Ross Wightman	983310d6a2	Fix #1935 , torch.amp.autocast -> torch.autocast (namespace issue for 1.10 pt compat)	2023-08-30 15:03:28 -07:00
Ross Wightman	300f54a96f	Another effcientvit (mit) tweak, fix torchscript/fx conflict with autocast disable	2023-08-20 15:07:25 -07:00
Ross Wightman	dc18cda2e7	efficientvit (mit) msa attention q/k/v ops need to be in float32 to train w/o NaN	2023-08-20 11:49:36 -07:00
Ross Wightman	7d7589e8da	Fixing efficient_vit torchscript, fx, default_cfg issues	2023-08-18 23:23:11 -07:00
Ross Wightman	c28324a150	Update efficient_vit (msra), hf hub weights	2023-08-18 16:45:37 -07:00
Ross Wightman	e700a32626	Cleanup of efficient_vit (mit), tweak eps for better AMP behaviour, formatting/cleanup, weights on hf hub	2023-08-18 16:06:07 -07:00
方曦	00f670fa69	fix bug in ci for efficientvits	2023-08-17 14:40:17 +08:00
方曦	047bab6ab2	efficientvit_mit stage refactor	2023-08-03 14:59:35 +08:00
方曦	43443f64eb	fix efficientvits	2023-08-02 14:12:37 +08:00
方曦	82d1e99e1a	add efficientvit(msra)	2023-08-01 18:51:08 +08:00
方曦	b91a77fab7	add EfficientVit (MIT)	2023-08-01 12:42:21 +08:00