e2b72f5da0a45f9bc81fd7f926a1a6205bb3b182,models/base_model.py,BaseModel,load_networks,#BaseModel#Any#,75

Before Change

                save_filename = "%s_net_%s.pth" % (which_epoch, name)
                save_path = os.path.join(self.save_dir, save_filename)
                net = getattr(self, "net" + name)
                net.module.load_state_dict(torch.load(save_path))

    // print network information
    def print_networks(self, verbose):
        print("---------- Networks initialized -------------")

After Change

                save_filename = "%s_net_%s.pth" % (which_epoch, name)
                save_path = os.path.join(self.save_dir, save_filename)
                net = getattr(self, "net" + name)
                if len(self.gpu_ids) > 0 and torch.cuda.is_available():
                    net.module.load_state_dict(torch.load(save_path))
                else:
                    net.load_state_dict(torch.load(save_path))

    // print network information
    def print_networks(self, verbose):
        print("---------- Networks initialized -------------")
        for name in self.model_names:
            if isinstance(name, str):

Instances