SLANTbrainSeg

SLANTbrainSeg copied to clipboard

Not using GPU

1

Hi I am running this docker image on CUDA 11 machine. I am able to see GPU information by doing nvidia-smi inside the docker. But when running inference it is...

AkhilaPerumalla123

training

1

Hi When I use CPU I get this : /extra/python/torchsrc/trainer.py:192: UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead. and there is no progressing .....

liamirpy

The model encoder has batch normalization enabled in 2D(`nn.BatchNorm2d`) instead it should be 3D?

ranabanik

Hello, while working on a computer vision task to classify parkinson diseases, I found that a more accurate brain mri segmentation mask was necessary. I originally relied on the fastsurfer...

penguin1109