PyTorch
model.pt
model.pth
commandpath = os.path.join(model_dir, "model.pth") torch.save(model.cpu().state_dict(), path)
code.path = os.path.join(model_dir, "model.pt") torch.jit.save(model, path)
<Video here>
Tensorflow