          self.serializer = Serializer.get_serializer("json")
          filename = "{}_state.{}".format(model_name, self.serializer.ext)
          self.filename = str(model_dir / filename)
          self.iterations = 0
          self.session_iterations = 0
          self.training_size = training_image_size
          self.sessions = dict()
          self.lowest_avg_loss = dict()
          self.inputs = dict()
 
</s> remove                 state = {"sessions": self.sessions,
 </s> add                 state = {"name": self.name,
                         "sessions": self.sessions,
        self.name = model_name
          self.serializer = Serializer.get_serializer("json")
          filename = "{}_state.{}".format(model_name, self.serializer.ext)
          self.filename = str(model_dir / filename)
        self.name = model_name
          self.iterations = 0
          self.session_iterations = 0
          self.training_size = training_image_size
          self.sessions = dict()
          self.lowest_avg_loss = dict()
          self.inputs = dict()
keep keep add keep keep keep keep keep keep
 Add model name to state file 
plugins/train/model/_base.py