toDo. See if we would like to implement mixed precision. Could improve performance / speed https://towardsdatascience.com/optimize-pytorch-performance-for-speed-and-memory-efficiency-2022-84f453916ea6 https://pytorch.org/blog/what-every-user-should-know-about-mixed-precision-training-in-pytorch/
toDo. See if we would like to implement mixed precision.
Could improve performance / speed
https://towardsdatascience.com/optimize-pytorch-performance-for-speed-and-memory-efficiency-2022-84f453916ea6
https://pytorch.org/blog/what-every-user-should-know-about-mixed-precision-training-in-pytorch/