feat: save intermediate trained models during challenge + demo

[sameerashahh]

I save the models that have been trained when challenge runs and, provide a demo for how you can load the saved model and plot the test results of that saved model.