Here we used the kaggle dataset to train our model , you will saved the model after you have trained it .Feel free to add checkpoints in between training epochs.
script is for testing the model after you have trained and saved the model . In test script we load that model and test it in differnet maps