Model Defination code : https://gist.github.com/81dff17eb270594637b5b356591bf8c1.git 16
I converted the model to Torch Script via tracing as shown in the tutorial
https://pytorch.org/tutorials/advanced/cpp_export.html#step-4-executing-the-script-module-in-c 25