search

NJU-LHRS / official-CMID

The official implementation of paper "Unified Self-Supervised Learning Framework for Remote Sensing Images".
80 stars 5 forks source link

MLP中出现了nn.Batchnorm2d？ #25

Open swjtulinxi opened 1 month ago

swjtulinxi commented 1 month ago

class MLP(nn.Module): def init(self, in_dim: int = 256, hidden_dim: int = 512, out_dim: int = 256, norm_layer=nn.BatchNorm1d, act_layer=nn.ReLU, drop: float = 0.):

MLP中出现了nn.batchnorm2d和nn.relu，这两个不是用于卷积的吗，可以用到nn.liner后面，不会出现维度错误？

pUmpKin-Co commented 1 month ago

您好.不会的,BatchNorm1D可以用于Sequence-Wise Operation. 同时最普通的1x1 Conv=Linear Layer.