Link to the colab notebook reproducing the issue I am facing wrt forward pass with custom model is attached below.
One question about forward method in the BasicBlock class. When and how it triggered? Somehow automatically by triggering the Resnet.Forward method?
Finally, my understanding of the Resnet here: a blog post and three videos. Thanks @amanarora
3 Likes
did you end up solving your memory issue?