Use jit.trace to convert the model to a mobile-deployable model

My pytorch_lite version is 1.9, when I convert with 1.9 version of pytorch, I get the following error:

transforms.RandAugment was added in torchvision==0.12 so you might need to update your torchvison (and PyTorch) installation.