Hello,

I'm trying to implement data parallel in libkge by integrating the _CustomDataParallel from your code, but it doesn't seem to work as expected. The data parallelism does not function properly.

Could you please advise if there are any specific steps or considerations for enabling data parallelism with _CustomDataParallel in libkge?

Thank you.