Jen Wei mengalami spiral kematian learning rate saat menguji AdamW, Muon, dan implementasi terbaru Dion3 di dalam OLMo-core untuk sebuah talk Konferensi PyTorch mendatang.

  • Proses debugging mengungkapkan masalah yang melibatkan state optimizer terdistribusi dan penjadwalan LR.
  • Sebuah jebukan aliasing tensor PyTorch tertentu diidentifikasi sebagai penyebab utama ketidakstabilan.
  • Wei mendokumentasikan seluruh petualangan debugging tersebut di Medium untuk membantu orang lain menghindari jebukan memori serupa di optimizer terdistribusi.