          if should_backup:
              self.backup()
          try:
              with open(self.filename, "wb") as out:
                  state = {"sessions": self.sessions,
                           "lowest_avg_loss": self.lowest_avg_loss,
                           "iterations": self.iterations,
                           "inputs": self.inputs,
                           "training_size": self.training_size,
                           "config": _CONFIG}
 
</s>
                state = {"name": self.name,
                         "sessions": self.sessions,
          if should_backup:
              self.backup()
          try:
              with open(self.filename, "wb") as out:
                state = {"name": self.name,
                         "sessions": self.sessions,
                           "lowest_avg_loss": self.lowest_avg_loss,
                           "iterations": self.iterations,
                           "inputs": self.inputs,
                           "training_size": self.training_size,
                           "config": _CONFIG}
keep keep keep keep replace keep keep keep keep keep
 Add model name to state file
plugins/train/model/_base.py