search

ViTAE-Transformer / ViTAE-Transformer-Remote-Sensing

A comprehensive list [SAMRS@NeurIPS'23, RVSA@TGRS'22, RSP@TGRS'22] of our research works related to remote sensing, including papers, codes, and citations. Note: The repo for [TGRS'22] "An Empirical Study of Remote Sensing Pretraining" has been moved to: https://github.com/ViTAE-Transformer/RSP
438 stars 53 forks source link

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

Open Akinpzx opened 1 year ago

Akinpzx commented 1 year ago

作者您好，我在尝试复现您在论文中，在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 commented 1 year ago

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

python setup.py installl

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

Akinpzx commented 1 year ago

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

Akinpzx commented 1 year ago

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

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

Akinpzx commented 1 year ago

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

DotWang commented 1 year ago

8

Akinpzx commented 1 year ago

能否看看这是为什么呢？ image

DotWang commented 1 year ago

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