itisianlee

``` class CLSHead(nn.Module): def __init__(self, in_ch=384, out_ch=1000, dropout=0.2): super().__init__() self.pool = nn.AdaptiveAvgPool2d(1) self.bn = nn.BatchNorm2d(in_ch) self.dropout = nn.Dropout(p=dropout, inplace=True) self.l = nn.Linear(in_ch, out_ch) def forward(self, x): x = self.bn(x) x...