Fix device selection for cuda

LSSTDESC / parsnip

Deep generative modeling of astronomical transient light curves

MIT License

7 stars 10 forks source link

Fix device selection for cuda #12

Closed hombit closed 2 years ago

hombit commented 2 years ago

I'm sorry, but my previous PR #8 introduced a bug on detecting CUDA devices: torch.backends.cuda is different from other backends and doesn't have is_available() function. This PR fixes it, I believe I properly tested it this time