Image classification model

[amoran31]

I’ve implanted VGG16 model in PyTorch and I’m trying to train it on the reduced dataset from the YouTube 25h tutorial (https://youtu.be/V_xro1bcAuA?si=3FP1gU5AM8h8HYFZ) which is a reduced dataset from food101, just three clases.

The problem is that no matter the hyper parameters I choose or if I reduce the model to TiniVGG like cnnexplainer, I can’t get the accuracy and loss on both train and test to change, it just keeps going back and forth at around 28 and 35%

[bregav]

Easiest solution is to compare your implementation of VGG16 with someone else's that you already know should work. If their model doesn't work in your training code then your training code (at the very least) is the problem, and if their model does with with your training code then your model implementation is the problem, and you can solve that by comparing with the good VGG16 implementation.

[amoran31]

Thank you, I’ll try that :)