Just updated the file to also include a vgg model, hope that helps ( also shows that the data is loaded correctly ( i had a few doubts when I saw the accuracy with resnet))
file is now here
Just updated the file to also include a vgg model, hope that helps ( also shows that the data is loaded correctly ( i had a few doubts when I saw the accuracy with resnet))
file is now here