Function _conv1d1x1

models/AxialLob/axiallob.py:7–9

(in_channels, out_channels)

def _conv1d1x1(in_channels, out_channels):
    return nn.Sequential(nn.Conv1d(in_channels, out_channels, kernel_size=1, stride=1, bias=False),
                         nn.BatchNorm1d(out_channels))


class GatedAxialAttention(pl.LightningModule):