Any code advice how to do that ?
When I trained my model I moved it to the cpu and then extracted with this code:
model.cpu()
example = torch.rand(1, 3, 224, 224)
traced_script_module = torch.jit.trace(model, example)
traced_script_module.save("app/src/main/assets/model.pt")