9 Commits

Author	SHA1	Message	Date
Pepijn	a66b50d372	scale lr decay if we reduce steps	2025-10-14 15:59:46 +02:00
Pepijn	4170d1b6f1	cleanup	2025-10-14 14:48:18 +02:00
Pepijn	4061b3f5b3	always use accelerate	2025-10-14 14:24:55 +02:00
Pepijn	50ff388bf6	update docs, and small improvements in train	2025-10-14 13:31:52 +02:00
Pepijn	c711a628b9	add tests	2025-10-13 16:25:46 +02:00
Pepijn	b65172f819	only log in main process	2025-10-10 14:05:53 +02:00
Pepijn	771b03c30d	fix pre commit	2025-10-10 13:35:26 +02:00
Pepijn	95b6035baa	Place logging under accelerate and update docs	2025-10-10 11:25:53 +02:00
Pepijn	4b7cd7211a	add docs and only push model once	2025-10-09 15:11:47 +02:00