894a2b921ea0ba9969cec62b484d03ab706ad25a,hypergan/configurable_component.py,ConfigurableComponent,layer_linear,#ConfigurableComponent#Any#Any#Any#,364

Before Change

    def layer_linear(self, net, args, options):

        if "*" in str(args[0]):
            reshape = [int(x) for x in args[0].split("*")]
            size = reduce(operator.mul, reshape)
        else:
            size = int(args[0])
            reshape = None
        net = self.ops.reshape(net, [self.ops.shape(net)[0], -1])

        options["reshape"] = reshape
        net = self.do_ops_layer(self.ops.linear, net, size, options)

        return net

After Change

    def layer_linear(self, net, args, options):

        self.nn_layers.append(nn.Flatten())//TODO only if necessary
        self.nn_layers.append(nn.Linear(self.current_input_size, int(args[0])))

    def layer_reshape(self, net, args, options):

Instances