4 Commits

Author	SHA1	Message	Date
Tri Dao	cadfa396b8	[Docker] Set torchmetrics==0.10.3	2022-12-30 02:42:28 -08:00
Tri Dao	984d5204e2	Update training Dockerfile to use flash-attn==0.2.6	2022-12-29 15:12:33 -08:00
Tri Dao	4a6eaa9f27	Update configs, add results	2022-11-29 04:46:43 -08:00
Tri Dao	0bf5e50038	Release training code	2022-11-28 17:34:40 -08:00