From c23abae29c6a9222114cdfaccc9c63491ee88c1d Mon Sep 17 00:00:00 2001 From: vince62s Date: Tue, 24 Dec 2024 12:35:13 +0100 Subject: [PATCH] clear cache after translation during scoring --- eole/trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/eole/trainer.py b/eole/trainer.py index 16f5cf16..c136f4d7 100644 --- 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()):