Yes, the forward method seems to return a list, so you would have to check why this is the case (check the type of x and try to isolate where it’s transformed to a list) and/or create a tensor, if needed.