https://versaaihub.com/modifying-gradient-accumulation/