def forward(self, x):
    main = self.main_max1(x)
    ext = self.ext_conv1(x)
    ext = self.ext_conv2(ext)
    ext = self.ext_conv3(ext)
    ext = self.ext_regul(ext)
    out = self.out_prelu((main + ext))
    return out
def forward(self, x):
    main = self.main_max1(x)
    ext = self.ext_conv1(x)
    ext = self.ext_conv2(ext)
    ext = self.ext_conv3(ext)
    ext = self.ext_regul(ext)
    out = self.out_prelu((main + ext))
    return out