Relative Content

Tag Archive for deep-learningpytorch

pytorch VGG3D model is not training

I’m training a model using pytorch on 3d MR brain images.
I have defined a few functions to handle training and evaluation but it seems like the model is not training at all.
The loss changes but the accuracy remains the same. precision and recall metrics are always zero
I can’t figure out what I’m doing wrong here.
My model is based on VGG16 but customized for 3D inputs.
The task is binary classification and there is a sigmoid layer at the end of model pipeline.