When mlp_ratio!=1, it errors

[Zhaoyi-Yan]

C keeps changing after processed by fcs. So you cannot use the same C here.

U-KAN/Seg_UKAN/archs.py

Lines 220 to 228 in 432a663

	x = self.fc1(x.reshape(B*N,C))
	x = x.reshape(B,N,C).contiguous()
	x = self.dwconv_1(x, H, W)
	x = self.fc2(x.reshape(B*N,C))
	x = x.reshape(B,N,C).contiguous()
	x = self.dwconv_2(x, H, W)
	x = self.fc3(x.reshape(B*N,C))
	x = x.reshape(B,N,C).contiguous()
	x = self.dwconv_3(x, H, W)