Does not converge

[zephyrize]

我就直接中文了。。。 你这个代码很奇怪，我一开始做3D的图像分割，把你的AxialAttention部分换成了自己的3D的，然后也跑起来了，虽然效果不好，但是至少能学到一些东西，loss也是下降的。

后来我去跑2D的，直接用源码跑，图像大小为256*256, channel是1. 结果竟然不收敛，网络学不到一点东西。