Closed M-R-Schaefer closed 5 months ago
Added EMA parameters, replaced my custom SAM implementation with the one in optax (faster and probably more correct) and added a cyclic cosine lr schedule.
todo
Added EMA parameters, replaced my custom SAM implementation with the one in optax (faster and probably more correct) and added a cyclic cosine lr schedule.
todo