              # TODO load is done in __init__ => look how to swap if possible
          return True
  
      def save_weights(self):
          self.netGA.save_weights(str(self.model_dir / netGAH5))
          self.netGB.save_weights(str(self.model_dir / netGBH5))
          self.netDA.save_weights(str(self.model_dir / netDAH5))
          self.netDB.save_weights(str(self.model_dir / netDBH5))
          print ("Models saved.")
 
</s> remove         print ("Models saved.") </s> add  </s> remove     def __init__(self, model_dir):
 </s> add     def __init__(self, model_dir, gpus):

        if self.gpus > 1:
            self.netGA_sm.save_weights(str(self.model_dir / netGAH5))
            self.netGB_sm.save_weights(str(self.model_dir / netGBH5))
        else:
            self.netGA.save_weights(str(self.model_dir / netGAH5))
            self.netGB.save_weights(str(self.model_dir / netGBH5))

              # TODO load is done in __init__ => look how to swap if possible
          return True
  
      def save_weights(self):
        if self.gpus > 1:
            self.netGA_sm.save_weights(str(self.model_dir / netGAH5))
            self.netGB_sm.save_weights(str(self.model_dir / netGBH5))
        else:
            self.netGA.save_weights(str(self.model_dir / netGAH5))
            self.netGB.save_weights(str(self.model_dir / netGBH5))
        if self.gpus > 1:
            self.netGA_sm.save_weights(str(self.model_dir / netGAH5))
            self.netGB_sm.save_weights(str(self.model_dir / netGBH5))
        else:
            self.netGA.save_weights(str(self.model_dir / netGAH5))
            self.netGB.save_weights(str(self.model_dir / netGBH5))
          self.netDA.save_weights(str(self.model_dir / netDAH5))
          self.netDB.save_weights(str(self.model_dir / netDBH5))
          print ("Models saved.")
keep keep keep keep replace replace keep keep keep
 Fix broken multigpu GAN loading. (#280)

* Fix broken multigpu GAN loading.

Fix for loading into the multi GPU model when it needs to load/save the original model.

* Move reference change.

Move reference change for single model so it is defined before load

* Add mutliGPU support to GAN128

Added support for mutligpu to GAN128 
plugins/Model_GAN/Model.py