Solved by ptrblck in post #2 
              
              
                Based on this error message it seems you are passing a list of tensors to the model while a tensor is expected so you would have to check the inputs to the model.