amirid

@siarez @RohitKeshari I added the following lines to the torchsample/functions/affine.py file and it works. ` if cuda: coords = Variable(th_iterproduct(x.size(1),x.size(2),x.size(3)).float(), requires_grad=False).cuda() else: coords = Variable(th_iterproduct(x.size(1),x.size(2),x.size(3)).float(), requires_grad=False) ` Find the function...

Hi, I'm using Python3.6, CUDA 9.2, Pytorch 1.2.