          for network in self.networks.values():
              if not network.side:
                  is_loaded = network.load(predict=self.predict)
              else:
                  is_loaded = network.load(fullpath=model_mapping[network.side][network.type],
                                           predict=self.predict)
              if not is_loaded:
                  break
          if is_loaded:
              logger.info("Loaded model from disk: '%s'", self.model_dir)
          return is_loaded
 
</s> remove                 is_loaded = network.load(predict=self.predict)
 </s> add                 is_loaded = network.load() </s> remove     def load(self, fullpath=None, predict=False):
 </s> add     def load(self, fullpath=None): </s> remove             if predict:
                raise ValueError("Unable to load training data. Error: {}".format(str(err)))
 </s> add 
                is_loaded = network.load(fullpath=model_mapping[network.side][network.type])
          for network in self.networks.values():
              if not network.side:
                  is_loaded = network.load(predict=self.predict)
              else:
                is_loaded = network.load(fullpath=model_mapping[network.side][network.type])
                is_loaded = network.load(fullpath=model_mapping[network.side][network.type])
              if not is_loaded:
                  break
          if is_loaded:
              logger.info("Loaded model from disk: '%s'", self.model_dir)
          return is_loaded
keep keep keep keep replace replace keep keep keep keep keep
 Fix VRAM issue with reloading models 
plugins/train/model/_base.py