Convert saved model state dict to jit trace model

[Skyworker (Christian)]

Is there a loss in doing this? Or would this work?