Model saving and loading

[kylase]

Saving and Loading model

Reference: PyTorch

Applications

• Saving/loading of model for Tensorboard Integration and logging #17 as we want to monitor how the performance of the model against validation/test data progresses over the training pass/epoch.
• For inference with Flask (REST API)