4a6cb19fc21993fe0505f3962da6bdfa94de4a0a,cube/models/vocoder.py,BeeCoder,load,#BeeCoder#,85

Before Change

        x = 0

    def load(self, output_base):
        self.network.load_state_dict(torch.load(output_base + ".network"))
        self.network.to(device)
        // self.model.populate(output_base + ".network")

    def _predict_one(self, mgc, noise):

After Change

        x = 0

    def load(self, output_base):
        if torch.cuda.is_available():
            self.network.load_state_dict(torch.load(output_base + ".network"))
        else:
            self.network.load_state_dict(
                torch.load(output_base + ".network",  map_location=lambda storage, loc: storage))
        self.network.to(device)
        // self.model.populate(output_base + ".network")

    def _predict_one(self, mgc, noise):

Instances