          self.encoder_dim = encoder_dim
          self.trainer = trainer
  
          self.state = State(self.model_dir, self.name, no_logs, training_image_size)
          self.rename_legacy()
          self.load_state_info()
  
          self.networks = dict()  # Networks for the model
          self.predictors = dict()  # Predictors for model
 
</s> remove         K.clear_session()
 </s> add         if not self.is_legacy:
            K.clear_session()
        self.is_legacy = False
          self.encoder_dim = encoder_dim
          self.trainer = trainer
  
          self.state = State(self.model_dir, self.name, no_logs, training_image_size)
        self.is_legacy = False
          self.rename_legacy()
          self.load_state_info()
  
          self.networks = dict()  # Networks for the model
          self.predictors = dict()  # Predictors for model
keep keep keep add keep keep keep keep keep
 bugfix: Don't clear session when loading legacy weights 
plugins/train/model/_base.py