Method __init__

toolkit/chelper.py:9–12

(self, model)

class augment_inputs_network(nn.Module):
    def __init__(self, model):
        super(augment_inputs_network, self).__init__()
        self.model = model
        self.adapter = nn.Conv2d(in_channels=6, out_channels=3, kernel_size=3, stride=1, padding=1)

    def forward(self, x):
        x = self.adapter(x)