organize models, rename MMT take two

lium-lst · Feb 20, 2020 · 117a359 · 117a359
1 parent 870d556
commit 117a359
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git 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 → nmtpytorch/models/simple_mmt.py b/nmtpytorch/models/mnmt.py → 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)