fairseqのinteractiveをクラス化する - Qiita

やりたいこと fairseqで学習したモデルの読み込みが長いので、fairseq-interactiveと同じ動作をするclassを作成する。 fairseqのバージョンは0.9.0を使っている。 コード from collections import namedtuple import torch from fairseq import checkpoint_utils, options, tasks, utils from fairseq.data import encoders Batch = namedtuple('Batch', 'ids src_tokens src_lengths') Translation = namedtuple('Translation', 'src_str hypos pos_scores alignments') def make_batches(l