def __init__(self, sz: Optional[int] = None):
        "Output will be 2*sz or 2 if sz is None"
        super().__init__()
        sz = sz or 1
        self.ap, self.mp = nn.AdaptiveAvgPool1d(sz), nn.AdaptiveMaxPool1d(sz)
Output will be 2*sz or 2 if sz is None
analyze/convnexts4nd/src/models/baselines/nonaka/basic_conv1d.py