changes

Helsinki-NLP · Oct 7, 2024 · 70bc59d · 70bc59d
1 parent b234e2e
commit 70bc59d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/mammoth/model_builder.py b/mammoth/model_builder.py
@@ -8,7 +8,7 @@
 from functools import partial
 from torch.nn.init import xavier_uniform_
 from typing import Optional, List, Dict, Tuple
-from mammoth.modules.x_transformers import TransformerWrapper
+from mammoth.modules.x_tf import TransformerWrapper
 from x_transformers.x_transformers import TokenEmbedding
 
 from mammoth.distributed.components import (

diff --git a/mammoth/modules/x_transformers.py → mammoth/modules/x_tf.py b/mammoth/modules/x_transformers.py → mammoth/modules/x_tf.py
diff --git a/mammoth/trainer.py b/mammoth/trainer.py
@@ -382,11 +382,11 @@ def validate(self, valid_iter, moving_average=None, task=None):
             for batch, metadata, _ in valid_iter:
                 if stats is None:
                     stats = mammoth.utils.Statistics()
-
+                
                 src, src_lengths = batch.src if isinstance(batch.src, tuple) else (batch.src, None)
                 decoder_input = batch.tgt[:-1]
                 target = batch.tgt[1:]
-
+                
                 with torch.cuda.amp.autocast(enabled=self.optim.amp):
                     # F-prop through the model.
                     logits, decoder_output = valid_model(