TransUNet

TransUNet代码调试记录

bug

1.运行测试代码test.py

AttributeError: “‘skip_channels’”

找到的网上有关错误的信息：skip_channels

按提示把vit_seg_modeling.py中的

skip_channels = self.config.skip_channels

改为

skip_channels = [512,256,64,16]

然后又有另一个bug：

看了一下我的目录结构，没有epoch_29.pth：

找到解决办法：Change the max_epochs, batch_size as you used when training.

-------------------------------------------分界线------------------------------------

不知道这两个参数要改成什么样，卡住了。。。。。。。。

2.test.py成功运行

参考这位大佬的回答修改终于成功了！nice!

一定要保证test.py里的max_epoches和batch_size参数是跟train.py相同的，我用他在终端调用train的命令调用不了，直接运行了train.py，那各个参数就是他代码里的默认参数，作者的test.py和train.py的max_epoches和batch_size参数不一样，要改成一样的。Swin-UNet也是一样的道理。

还有vit_name预训练模型也要用一样的，代码里的默认模型是不一样的。