894a2b921ea0ba9969cec62b484d03ab706ad25a,hypergan/configurable_component.py,ConfigurableComponent,layer_conv,#ConfigurableComponent#Any#Any#Any#,356

Before Change

        else:
            channels = self.ops.shape(net)[-1]

        return self.do_ops_layer(self.ops.conv2d, net, channels, options)

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

        if "*" in str(args[0]):

After Change

            self.set_layer(args[0], net)
        return net

    def layer_conv(self, net, args, options):
        if len(args) > 0:
            channels = int(args[0])
        else:
            channels = self.ops.shape(net)[-1]

        options = hc.Config(options)

        self.nn_layers.append(nn.Conv2d(self.current_channels, channels, options.filter or 3, options.stride or 2, (options.filter or 3)//2))
        self.nn_layers.append(nn.ReLU())//TODO
        self.current_channels = channels
        self.current_width = self.current_width // 2 //TODO
        self.current_height = self.current_height // 2 //TODO
        self.current_input_size = self.current_channels * self.current_width * self.current_height

Instances