StreamingASR/save_model_for_mobile.py

import torch
import torchaudio
from torch.utils.mobile_optimizer import optimize_for_mobile

def get_demo_wrapper():
    wrapper = torch.jit.load("scripted_wrapper_tuple_no_transform.pt")
    return wrapper

wrapper = get_demo_wrapper()
scripted_model = torch.jit.script(wrapper)
optimized_model = optimize_for_mobile(scripted_model)
optimized_model._save_for_lite_interpreter("streaming_asr.ptl")
print("Done _save_for_lite_interpreter")