All Questions
2 questions
2
votes
1
answer
5k
views
How to reduce overfitting in a pre-trained network
I have a custom dataset with 10 classes and I am using a pre-trained resnet18 model from torch-vision. I can clearly see it's over-fitting because: the model is trained for 75 epochs with a batch size ...
- 143
1
vote
1
answer
1k
views
trying to decrease overfitting with regularisation in CNN
I am doing transfer learning by retraining the publicly available inception layer, without regularisation here are my initial parameters and results:
...
- 161