          if self.side:
              name += "_{}".format(self.side)
          return name
  
      def load(self, fullpath=None, predict=False):
          """ Load model """
          fullpath = fullpath if fullpath else self.filename
          logger.debug("Loading model: '%s'", fullpath)
          try:
              network = load_model(self.filename, custom_objects=get_custom_objects())
 
</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                 is_loaded = network.load(predict=self.predict)
 </s> add                 is_loaded = network.load() </s> remove             if predict:
                raise ValueError("Unable to load training data. Error: {}".format(str(err)))
 </s> add 
    def load(self, fullpath=None):
          if self.side:
              name += "_{}".format(self.side)
          return name
  
    def load(self, fullpath=None):
          """ Load model """
          fullpath = fullpath if fullpath else self.filename
          logger.debug("Loading model: '%s'", fullpath)
          try:
              network = load_model(self.filename, custom_objects=get_custom_objects())
keep keep keep keep replace keep keep keep keep keep
 Fix VRAM issue with reloading models 
plugins/train/model/_base.py