A generated training loop 'works' but the loss curve goes strange after the first few steps, and printing w.grad shows it growing every iteration even when the loss doesn't. Which bug is this?
A generated training loop 'works' but the loss curve goes strange after the first few steps, and printing w.grad shows it growing every iteration even when the loss doesn't. Which bug is this?