              logger.warning("Failed loading existing training data. Generating new models")
              logger.debug("Exception: %s", str(err))
              return False
          except OSError as err:  # pylint: disable=broad-except
              if predict:
                  raise ValueError("Unable to load training data. Error: {}".format(str(err)))
              logger.warning("Failed loading existing training data. Generating new models")
              logger.debug("Exception: %s", str(err))
              return False
          self.network = network  # Update network with saved model
          self.network.name = self.type
 
</s> remove             if predict:
                raise ValueError("Unable to load training data. Error: {}".format(str(err)))
 </s> add  </s> remove                 is_loaded = network.load(fullpath=model_mapping[network.side][network.type],
                                         predict=self.predict)
 </s> add                 is_loaded = network.load(fullpath=model_mapping[network.side][network.type]) </s> remove     def load(self, fullpath=None, predict=False):
 </s> add     def load(self, fullpath=None):
              logger.warning("Failed loading existing training data. Generating new models")
              logger.debug("Exception: %s", str(err))
              return False
          except OSError as err:  # pylint: disable=broad-except


              logger.warning("Failed loading existing training data. Generating new models")
              logger.debug("Exception: %s", str(err))
              return False
          self.network = network  # Update network with saved model
          self.network.name = self.type
keep keep keep keep replace replace keep keep keep keep keep
 Fix VRAM issue with reloading models 
plugins/train/model/_base.py