To avoid overfitting there are a few things you can do :
Create custom convnet architecture using
The model is cut according to
cut and it may be
pretrained, in which case, the proper set of weights is downloaded then loaded.
init is applied to the head of the model, which is either created by
y_range) or is
- finally if there is no way to get it to regularise with all this, you still have the option to use resnet18
Hope it helps !