Does DGL work with pytorch lightning + mixed precision?

Hi all, I had a good experience using pytorch lightning for another project. I wanted to know if someone has tried it with PL, before i give it a try.

On top of that, do you guys think that PL to fp16 would work, or would I have to follow the specific directions here, compiling from source, etc: Chapter 8: Mixed Precision Training — DGL 0.6.1 documentation


I just pushed a PR for single-GPU training with PyTorch Lightning in [Model] Training GraphSAGE with PyTorch Lightning by BarclayII · Pull Request #2878 · dmlc/dgl · GitHub.

The fp16 support is currently incomplete; it supports GPU if you compile it from source, and it does not support CPU yet.

1 Like