Solved by InnovArul in post #2 
              
              
                Can you make sure to post the complete error text with full error message? 
Also,  if you are using GPU,  make sure both data and model are in GPU while calling forward method.