def load(self): if self.load_path_G is not None: print('loading model for G [%s] ...' % self.load_path_G) load_network(self.load_path_G, self.netG) if self.load_path_D is not None: print('loading model for D [%s] ...' % self.load_path_D) load_network(self.load_path_D, self.netD)
def load(self): if self.load_path_sftnet is not None: print('loading model for SFTNET [%s] ...' % self.load_path_sftnet) load_network(self.load_path_sftnet, self.sft_net)