the base model located in notebooks/bolt-mini-base.ipynb seems to be overfitted as the training loss is very low but validation loss is sky rocketing. Similar situation goes on with training accuracy and validation accuracy.
Note : The present base model is currently trained only on a subset of the intel dataset specifically only "seg_train" directory has been taken into account and further updates will be made to accommodate remaining data and also include multiple other datasets through a simple pipeline that will merge multiples datasets together for preprocessing.
Possible Fixes :-
1. Increase size of dataset
2. Add keras regularizers (l1,l2)
3. Use data augmentation [Optional]
the base model located in notebooks/bolt-mini-base.ipynb seems to be overfitted as the training loss is very low but validation loss is sky rocketing. Similar situation goes on with training accuracy and validation accuracy.
Note : The present base model is currently trained only on a subset of the intel dataset specifically only "seg_train" directory has been taken into account and further updates will be made to accommodate remaining data and also include multiple other datasets through a simple pipeline that will merge multiples datasets together for preprocessing.
Possible Fixes :-
1. Increase size of dataset
2. Add keras regularizers (l1,l2)
3. Use data augmentation [Optional]