7348eb260cb8be71a9d469f9a27a2625b0ce1dce,models/networks.py,SIGGRAPHGenerator,forward,#SIGGRAPHGenerator#,349

Before Change

        conv8_3 = self.model8(conv8_up)

        if(self.classification):
            return self.model_class(conv8_3)
        else:
            // embed()
            conv9_up = self.model9up(conv8_3) + self.model2short9(conv2_2)
            conv9_3 = self.model9(conv9_up)

After Change

        if(self.classification):
            out_class = self.model_class(conv8_3)

            conv9_up = self.model9up(conv8_3.detach()) + self.model2short9(conv2_2.detach())
            conv9_3 = self.model9(conv9_up)

            conv10_up = self.model10up(conv9_3) + self.model1short10(conv1_2.detach())
            conv10_2 = self.model10(conv10_up)

            out_reg = self.model_out(conv10_2)

            return (out_class, out_reg)
        else:
            conv9_up = self.model9up(conv8_3) + self.model2short9(conv2_2)
            conv9_3 = self.model9(conv9_up)
            conv10_up = self.model10up(conv9_3) + self.model1short10(conv1_2)

Instances