Great. Thank you. Will be glad to get your insights about the memory usage in your computer later (BTW-  will you please tell me how much memory usage does your computer show when the batch size is 4? ).
I am using Pytorch 0.4, but still using “Variable” just from the habit. I guess that the only thing that would change in the Pytorch 0.4 syntax is that I would have to delete the name “Variable” and just leave “img = img.cuda()”.
Thanks a lot again