search

Nikronic / CoarseNet

Modified version of U-net based on "Convolutional Networks for Biomedical Image Segmentation (Ronneberger et al., 2015)" paper.
https://arxiv.org/abs/1505.04597
MIT License
25 stars 2 forks source link

Add ability to detect GPU if exists and use proper device #2

Closed Nikronic closed 5 years ago

Nikronic commented 5 years ago