One of your datasets seem to return an int where a tensor is expected.
You could add print statements into the Dataset.__getitem__ methods, check the type of all returned objects, and make sure they are tensors.