Thank you for the reference.
I tried running the code from your original post, and got the following error:
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [64, 3, 7, 7], but got 3-dimensional input of size [3, 224, 224] instead
Since I can’t run the code, I can’t debug it either.