diff --git a/dan/manager/training.py b/dan/manager/training.py index fdaf5d3d2c2368bf52c639fd4c12ca20bcbbd925..8f4de234a5fe0391d999e12c21b175fa6dabfd25 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