Solved by ptrblck in post #2 
              
              
                Based on the error message the second argument (prediction_loader) is passed as the input to the model, which is a DataLoader instance and will thus fail. 
Pass tensors to the model by iterating the DataLoader instead.