search

Shreeyak / pytorch-lightning-segmentation-template

Semantic Segmentation on the LaPa dataset using Pytorch Lightning
Apache License 2.0
13 stars 5 forks source link

bugfix: pass number of gpus to Trainer instead of gpu ids. #15

Closed Shreeyak closed 3 years ago

Shreeyak commented 3 years ago

See: https://github.com/PyTorchLightning/pytorch-lightning/issues/5264

We now pass in number of gpus to use to PL.Trainer. Set CUDA_VISIBLE_DEVICES to set the gpu ids to use.