GitHub

关于扫描方向实现问题 #13

for layer in range (self.hnum): self.hhmamba.append (nn.ModuleList ( [ SS2D6 (d_model=dims, dropout=0, d_state=d_state, scan_type='lh'), PreNorm (dims, FeedForward (dim=dims)) ])) ...