fix: pickle error due to Lock object

tb_logger get's closed after training.
So we need to re-assign the new one, otherwise pickling fails.

closes #10034
This commit is contained in:
Matthias
2025-03-11 19:53:41 +01:00
parent d69b1566b1
commit b6a65ce125

View File

@@ -763,6 +763,8 @@ class IFreqaiModel(ABC):
init_model = None
else:
init_model = self.dd.model_dictionary[pair]
# Set "fresh" tb_logger - the one in model_dictionary has the writer closed.
init_model.tb_logger = self.tb_logger
return init_model