Skip to content

Instantly share code, notes, and snippets.

@jmp84
jmp84 / export.py
Created September 16, 2020 06:11
TorchScript MT model
import argparse
import logging
import torch
from fairseq.checkpoint_utils import load_model_ensemble_and_task
from fairseq.sequence_generator import SequenceGenerator
def get_args():
parser = argparse.ArgumentParser(