Solved by ynjiun_wang in post #2 
              
              
                Found the solution: change the following line setting to True: 
# Set to True if wanting to use torch.jit.script on a model
_SCRIPTABLE = True

in config.py