search

mateuszbuda / brain-segmentation-pytorch

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI
https://mateuszbuda.github.io/2017/12/01/brainseg.html
MIT License
718 stars 189 forks source link

fix hub loading on CPU-only machines #5

Closed soumith closed 5 years ago

soumith commented 5 years ago

this fixes the hub entry on non-cuda machines

mateuszbuda commented 5 years ago

Thanks!