  import keras
  from keras import losses
  from keras.models import load_model
  from keras.optimizers import Adam
  from keras.utils import get_custom_objects, multi_gpu_model
  
 
</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             if predict:
                raise ValueError("Unable to load training data. Error: {}".format(str(err)))
 </s> add  </s> remove             if predict:
                raise ValueError("Unable to load training data. Error: {}".format(str(err)))
 </s> add  </s> remove     def load(self, fullpath=None, predict=False):
 </s> add     def load(self, fullpath=None):
from keras import backend as K
  import keras
  from keras import losses
from keras import backend as K
  from keras.models import load_model
  from keras.optimizers import Adam
  from keras.utils import get_custom_objects, multi_gpu_model
  
keep add keep keep keep keep
 Fix VRAM issue with reloading models 
plugins/train/model/_base.py