1d50d1fbba5a48af58c1c5f5e732c207d58a7a87,hypergan/configurable_component.py,ConfigurableComponent,layer_linear,#ConfigurableComponent#Any#Any#Any#,375

Before Change

            nn.Linear(self.current_input_size, int(args[0]))
        ]
        if options.activation != "null":
            layers.append(nn.ReLU())//TODO

        self.current_input_size = int(args[0])
        return nn.Sequential(*layers)

After Change

            nn.Flatten(),
            nn.Linear(self.current_input_size, output_size)
        ]
        if len(shape) > 1:
            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))

        if options.activation != "null":
            layers.append(self.activation_to_module(options.activation or "relu"))

        self.current_input_size = output_size

Instances