Implement saving to Torchscript in the pytorch model #18
Description
In order to couple to FTorch we need to save the model as a Torchscript file/representation.
This could be done in a separate script, or as a method of the NN model, but ideally by using the tool provided in FTorch.
To do this we need the final model (at least the architecture if not the trained weights) so will discuss this with @yqsun91.
Examples, and a tool to save models can be found in https://github.com/Cambridge-ICCS/FTorch