<gh_stars>1-10
"""Module defining models."""
from onmt.models.model_saver import build_model_saver, ModelSaver
from onmt.models.model import NMTModel, ConvModel, KTransformerModel
from onmt.models.nmodel import NTransformerModel

__all__ = ["build_model_saver", "ModelSaver",
           "NMTModel", "check_sru_requirement", "ConvModel", "KTransformerModel", "NTransformerModel"]

"""Module defining models."""
from onmt.models.model_saver import build_model_saver, ModelSaver
from onmt.models.model import NMTModel, ConvModel, KTransformerModel
from onmt.models.nmodel import NTransformerModel

__all__ = ["build_model_saver", "ModelSaver",
           "NMTModel", "check_sru_requirement", "ConvModel", "KTransformerModel", "NTransformerModel"]
Module defining models.