It spits out torch.cuda.FloatTensor.
I solved the problem as you can see I had lot of firefox windows open. 100s of tabs already raised my memory to 60% after that if I start training Pytorch need an extra 30% that’s why my computer crashed.
Sorry to bother you
Thanks for the replies