7a65d9993bdda4a5248f6aa313abf687faa031a6,hypergan/configurable_component.py,ConfigurableComponent,layer_linear,#ConfigurableComponent#Any#Any#Any#,334

Before Change

        ]
        self.nn_init(layers[0], options.initializer)
        if len(shape) == 3:
            self.current_channels = shape[2]
            self.current_width = shape[0]
            self.current_height = shape[1]
            layers.append(Reshape(self.current_channels, self.current_height, self.current_width))

After Change

        layers += [nn.Linear(options.input_size or self.current_size.size(), output_size, bias=bias)]
        self.nn_init(layers[0], options.initializer)
        self.current_size = LayerSize(*reversed(shape))
        if len(shape) != 1:
            layers.append(Reshape(*self.current_size.dims))

        return nn.Sequential(*layers)

Instances