nnUNetV2修改笔记-D12

网络训练

周末把CAUNet的那一折搞好了,SAUNet跑不了,所以选择跳过去跑CABMUNet和Squeeze-and-Excitation去了。现在想了一个办法,就是先把一个网络重新复制一个预处理,把网络编号改一下
pre.webp
pre2.webp
然后修改plan文件,把里面的数据集编号和网络也改了
pre3.webp
pre4.webp
这样的话就可以同时处理好几个网络。跑出来的结果是这样的:
pre5.PNG这样的话不容易搞混,就算是文件夹名看不出来什么网络,点进去看一看plans.json也能看出来当初用的什么网络训练的。我觉得这样挺好的。

周末把CBAM和CA的训练结束了,SA的一直不收敛或者是梯度爆炸。之前把虚拟环境关了重开好像有效,现在正在跑SnE的第三折,等会把它跑完或者C了,把CBAM和CA先验证了再说,因为CBAM=CA+SA,我想的是把这三个先跑了看下效果。等会如果有进展再继续写。