Create a simple PyTorch Model and save it in Lite Format along with a single metadata file (model_info.txt)

CreateSimplePyTorchModelAndSaveInLiteFormat.py

import torch
import zipfile

class AddTensorsModel(torch.nn.Module):
  def __init__(self):
    super().__init__();

  def forward(self, x, y):
    return x + y

m = AddTensorsModel()
# Create a scripted mode so that we can save it for mobile consumption
scripted = torch.jit.script(m)

# Create an extra_files map so that we can add this additional
# metadata when saving the model.
extra_files = {
    "model_info.txt": "This model's forward() method accepts 2 tensors, and returns their sum",
}
scripted._save_for_lite_interpreter("AddTensorsModel.ptl", _extra_files = extra_files)