KeyError: "EncoderDecoder: 'ViTAE_Window_NoShift_basic is not in the models registry'"

[ZhiXin-Peng]

作者您好，我在尝试复现您在论文中，在Potsdam数据集上的实验时，运行以下代码
python -m torch.distributed.launch --nproc_per_node=1 --master_port=40001 tools/train.py configs/vitae_win/upernet_vitae_win_imp_window7_512x512_80k_potsdam.py --launcher 'pytorch'，
但是出现了KeyError: "EncoderDecoder: 'ViTAE_Window_NoShift_basic is not in the models registry'"的问题，请问这是为什么呢？有什么办法解决吗？
我还想请教一下如何正确复现您的实验呢？

[DotWang]

@Akinpzx 首先检查mmcv的类型，是不是能改那种的，其次看看有没有安装mmseg（你克隆下的这个文件夹其实就是mmseg，只不过可能不是最新版的）然后安装

python setup.py installl

安装完以后就可以复现了，potsdam需要自己准备，准备过程可以参考其他issue

[ZhiXin-Peng]

这是我安装的mmcv-full,您能看看有什么问题吗，我安装了mmseg之后仍然出现那个问题

[ZhiXin-Peng]

上面的问题我已经解决了，请问出现这个

我是需要在您的主页中下载这个吗

[ZhiXin-Peng]

我想要语义分割的预训练模型应该在哪找呢？

[DotWang]

#8

[ZhiXin-Peng]

能否看看这是为什么呢？

[DotWang]

@Akinpzx 没加 --cfg-options 'find_unused_parameters'=True