search

LuxImagingAI / DBSegment

This is a deep learning-based method to segment deep brain structures and a brain mask from T1 weighted MRI.
GNU General Public License v3.0
11 stars 6 forks source link

warning to be removed #1

Open MehriB opened 2 years ago

MehriB commented 2 years ago

Remove the following warning when run on CPU:

/mnt/lscratch/users/mbaniasadi2/dbsegment_test/env_cpu/lib/python3.7/site-packages/torch/cuda/amp/grad_scaler.py:115: UserWarning: torch.cuda.amp.GradScaler is enabled, but CUDA is not available. Disabling. warnings.warn("torch.cuda.amp.GradScaler is enabled, but CUDA is not available. Disabling.") /mnt/lscratch/users/mbaniasadi2/dbsegment_test/env_cpu/lib/python3.7/site-packages/torch/autocast_mode.py:141: UserWarning: User provided device_type of 'cuda', but CUDA is not available. Disabling warnings.warn('User provided device_type of \'cuda\', but CUDA is not available. Disabling')