7a65d9993bdda4a5248f6aa313abf687faa031a6,hypergan/configurable_component.py,ConfigurableComponent,layer_conv3d,#ConfigurableComponent#Any#Any#Any#,304

Before Change

        layers = [nn.Conv3d(options.input_channels or self.current_channels, channels, fltr, stride, padding = padding)]
        self.nn_init(layer, options.initializer)
        self.current_channels = channels
        if stride[1] > 1 or stride[2] > 1: //TODO
            self.current_width = self.current_width // stride[1] //TODO
            self.current_height = self.current_height // stride[2] //TODO
            print("conv", self.current_width, self.current_height, self.current_channels)
        self.current_input_size = self.current_channels * self.current_width * self.current_height
        return nn.Sequential(*layers)

    def layer_linear(self, net, args, options):
        options = hc.Config(options)

After Change

        layers = [nn.Conv3d(options.input_channels or self.current_size.channels, channels, fltr, stride, padding = padding)]
        self.nn_init(layer, options.initializer)
        self.current_size = LayerSize(frames, channels, self.current_size.height // stride[1], self.current_size.width // stride[2]) //TODO this doesn"t work, what is frames? Also chw calculation like conv2d
        return nn.Sequential(*layers)

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

Instances