Hello everyone,

I’m using tensorflow as my back-end and I’m interested in using tf.data.Dataset with DGL for speeding up the training.

I’m doing graph classification where at each iteration to graph changes therefore it has to be iterated over.

I’m guessing that because DGLGraph objects are arbitrary objects (instead of tensors or np.arrays) it makes using tf.data.Dataset or tf.keras.utils.Sequence challenging.

I would appreciate any help.