@user_123454321
Oh yeah… That was the issue here which I strive to fix. By the way, I’m new to Pytorch and it will be helpful to know whether is it normal to code like what I did (or) is there any other way to write custom Pytorch classes for Convolution, Relu, etc which utilizes GPU as native Pytorch classes do?