db8324230b2f99a589f49d13bd0950060646bf46,plugins/Model_GAN128/Model.py,GANModel,save_weights,#GANModel#,183

Before Change

        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))

After Change

        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))
        self.netDA.save_weights(str(self.model_dir / netDAH5))
        self.netDB.save_weights(str(self.model_dir / netDBH5))
        print ("Models saved.")

Instances