clear cache after translation during scoring

eole-nlp · Dec 24, 2024 · c23abae · c23abae
1 parent 1dbf345
commit c23abae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/eole/trainer.py b/eole/trainer.py
@@ -469,6 +469,7 @@ def validate(self, valid_iter, moving_average=None):
 
                 # Update statistics.
                 stats.update(metric_stats)
+                valid_model.decoder._clear_cache()
 
         if moving_average:
             for param_data, param in zip(model_params_data, self.model.parameters()):