I put this image together tonight as I was trying to internalize all the things that go on in a Resnet model. I am the kind of person who has to feel like I know something intimately to feel like I KNOW it at all. So this code and I are really good friends at this point.
I thought I would share in case this could help anyone else and their relationship with fully comprehending every step.
Have fun! And happy model training to you!
Special thanks to Aakash NS at his team at Jovian.ml!