Commit Graph

10 Commits

Author SHA1 Message Date
Pepijn f8a185f753 cleanup logging 2025-10-14 17:05:47 +02:00
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