diff --git a/nmtpytorch/models/__init__.py b/nmtpytorch/models/__init__.py index a305cb0f..3309416a 100644 --- a/nmtpytorch/models/__init__.py +++ b/nmtpytorch/models/__init__.py @@ -1,12 +1,17 @@ +##### +# NMT +##### from .nmt import NMT from .tfnmt import TransformerNMT -# MMT with FC-style global features -from .mnmt import MultimodalNMT - -# Spatial features + NMT +################ +# Multimodal NMT +################ +from .simple_mmt import SimpleMMT from .attentive_mmt import AttentiveMMT +############### # Speech models +############### from .asr import ASR from .multimodal_asr import MultimodalASR diff --git a/nmtpytorch/models/mnmt.py b/nmtpytorch/models/simple_mmt.py similarity index 99% rename from nmtpytorch/models/mnmt.py rename to nmtpytorch/models/simple_mmt.py index fa417744..9a454064 100644 --- a/nmtpytorch/models/mnmt.py +++ b/nmtpytorch/models/simple_mmt.py @@ -9,7 +9,7 @@ logger = logging.getLogger('nmtpytorch') -class MultimodalNMT(NMT): +class SimpleMMT(NMT): """A encoder/decoder enriched multimodal NMT. Integration types (feat_fusion argument)