A export pt and load to DJL example

[weican]

I've trained retinaFace https://github.com/biubug6/Pytorch_Retinaface in colab. I also converted the result of model pth file.
However, DJL shows the error:
torch::jit::load() received a file from torch.save(), but torch::jit::load() can only load files produced by torch.jit.save()
Does it have an example to guide me on how to train and export the correct pt model file?

[frankfliu]

DJL requires jit traced model. Please take a look: https://docs.djl.ai/master/docs/pytorch/how_to_convert_your_model_to_torchscript.html