7a65d9993bdda4a5248f6aa313abf687faa031a6,hypergan/configurable_component.py,ConfigurableComponent,layer_adaptive_avg_pool,#ConfigurableComponent#Any#Any#Any#,435

Before Change

        return Reshape(*dims)

    def layer_adaptive_avg_pool(self, net, args, options):
        print("adaptive start", self.current_width, self.current_height, self.current_channels)
        self.current_height //= 2
        self.current_width //= 2
        self.current_input_size = self.current_channels * self.current_width * self.current_height
        return nn.AdaptiveAvgPool2d([self.current_height, self.current_width])

    def layer_adaptive_avg_pool1d (self, net, args, options):
        self.current_height //= 2

After Change

        self.current_size = LayerSize(*dims)
        return Reshape(*dims)

    def layer_adaptive_avg_pool(self, net, args, options):
        self.current_size = LayerSize(self.current_size.channels, self.current_size.height // 2, self.current_size.width // 2)
        return nn.AdaptiveAvgPool2d([self.current_size.height, self.current_size.width])

    def layer_adaptive_avg_pool1d (self, net, args, options):
        self.current_size = LayerSize(self.current_size.channels, self.current_size.height // 2)

Instances