Solved by JuanFMontesinos in post #3 
              
              
                cast your input to float type by doing 
input=input.float() before calling the forward pass 
(I’m not sure if u should call this method after or before allocating tensor to cuda, before seems better.)