trained-models
trained-models copied to clipboard
Add PialNN
This is the model that needs to be integrated https://github.com/neuroneural/PialNN_fork
built docker image successfully
Ran into https://github.com/neuroneural/PialNN_fork/issues/1
@gaiborjosue This is why PialNN is hanging. The version of torch in the image supports only specific cards. I will try running this on a different card and keep you posted.
@hvgazula, I see. Great, let me know.
Works on tesla-v100.
Ready for testing in GH workflow. For more info, please refer to https://github.com/gaiborjosue/trained-models-fork/issues/64
testing model in singularity with --model
flag and it seems to be running fine. See below image