From e2235375a259a98ede0062ed25900f0f5c566466 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=A9lodie=20Boillet?= <boillet@teklia.com>
Date: Tue, 8 Aug 2023 12:37:54 +0000
Subject: [PATCH] Add condition for charset transfer
---
dan/manager/training.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/dan/manager/training.py b/dan/manager/training.py
index fdaf5d3d..8f4de234 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -297,6 +297,7 @@ class GenericTrainingManager:
if (
"end_conv" in key
and "transfered_charset" in self.params["model_params"]
+ and self.params["model_params"]["transfered_charset"]
):
self.adapt_decision_layer_to_old_charset(
model_name, key, checkpoint, state_dict_name
--
GitLab