  
      def load_models(self, swapped):
          """ Load models from file """
          logger.debug("Load model: (swapped: %s)", swapped)
          model_mapping = self.map_models(swapped)
          for network in self.networks.values():
              if not network.side:
                  is_loaded = network.load()
              else:
                  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                 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): </s> remove         #else:
 </s> add         # else:
        if not self.models_exist and not self.predict:
            logger.info("Creating new '%s' model in folder: '%s'", self.name, self.model_dir)
            return
        if not self.models_exist and self.predict:
            logger.error("Model could not be found in folder '%s'. Exiting", self.model_dir)
            exit(0)

        K.clear_session()
  
      def load_models(self, swapped):
          """ Load models from file """
          logger.debug("Load model: (swapped: %s)", swapped)
        if not self.models_exist and not self.predict:
            logger.info("Creating new '%s' model in folder: '%s'", self.name, self.model_dir)
            return
        if not self.models_exist and self.predict:
            logger.error("Model could not be found in folder '%s'. Exiting", self.model_dir)
            exit(0)

        K.clear_session()
          model_mapping = self.map_models(swapped)
          for network in self.networks.values():
              if not network.side:
                  is_loaded = network.load()
              else:
                  is_loaded = network.load(fullpath=model_mapping[network.side][network.type])
keep keep keep add keep keep keep keep keep keep
 Fix VRAM issue with reloading models 
plugins/train/model/_base.py