transfering of data to gpu with custom data type
❓ Questions and Help
What is your question?
The batch, that should be passed to the train/val step consists of a list of https://pytorch-geometric.readthedocs.io/en/latest/modules/data.html torch_geometric Data objects. The batch is not passed to the corresponding gpus. I assumed that this would happen in the pytorch_lightning/trainer/distrib_parts.py transfer methods, however the code never arrives in these methods. Where do I have to change the code of lightning to allow for custom data types ?
Code
What have you tried?
If I change the batch to a list of tensors, they are correctly transfered, so I assume the rest of my code works fine. Transfering the data by adding the corresponding to device calls in the train step also works.
What's your environment?

Ubuntu 18.04
pip
Version '0.7.1'
dgx2 cluster, no slurm