Hi, I have some questions here:
"
So after the first convolution using your formular, we will have
[3, 64, 198, 148]

After the second Conv2d operation, we will have
[3, 128, 196, 146].

"
Since we assume batch_size=36, maybe it should be that after the first convolution layer, we will have
[36, 64, 198, 148].

And after the second convolution layer, we will have
[36, 128, 196, 146].