You wrote your own Adam with , and both moment estimates initialized to zero, and the update divides by directly.
The loss spikes in the first few hundred steps at a learning rate that is stable with the library's Adam.
You wrote your own Adam with , and both moment estimates initialized to zero, and the update divides by directly.
The loss spikes in the first few hundred steps at a learning rate that is stable with the library's Adam.