comet-atomic-2020 icon indicating copy to clipboard operation
comet-atomic-2020 copied to clipboard
verified publisher icon allenai

Update PyTorch-Lightning Requirement to 1.6.4

Open dbl001 opened this issue 3 years ago • 0 comments

I'd like to use 'comet' with the new PyTorch 'mps' backend. The PyTorch 'mps' implementation is still being 'shaken-out', as is PyTorch-Lightning 1.6.4, however, some of the code works. Can you create a development branch of 'comet' that can use the latest version of 'PyTorch-Lightning' so we can run on the 'mps' backend (GPU).

dbl001 avatar Jun 17 '22 15:06 dbl001