News for hackers. Stuff that matters.
Re:Accelerated PyTorch Training on M1 Mac by dgacmu on Wednesday May 18, 2022 @03:01PM (#31426381)
You have to stash more information from the forward pass in order to calculate the gradients during backprop. You can't just naively use an inference accelerator as part of training - inference-only gets to discard intermediate activations immediately.

(Also, many inference accelerators use lower precision than you do when training)

There are tricks you can do to use inference to accelerate training, such as one we developed to focus on likely-poorly-performing examples: https://arxiv.org/abs/1910.00762

There are no comments on this comment yet.