Migrate code base to `pytorch`

NOEL-MNI / deepFCD

Automated Detection of Focal Cortical Dysplasia using Deep Learning

https://noel.bic.mni.mcgill.ca/projects/

BSD 3-Clause "New" or "Revised" License

8 stars 3 forks source link

Open ravnoor opened 2 years ago

ravnoor commented 2 years ago

Keras (w/ theano backend) doesn't support model weight conversion to other frameworks (pytorch, tensorflow, onnx, etc.)
Original theano has been deprecated
deepMask source code is already in pytorch
pytorch has native support for Mx Apple Silicon (cpu+gpu) and ROCm platforms