search

microsoft / unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
https://aka.ms/GeneralAI
MIT License
19.11k stars 2.44k forks source link

KeyError: 'BEiT is not in the backbone registry' #608

Open SamXiaosheng opened 2 years ago

SamXiaosheng commented 2 years ago

when build for segmentation task，occur the KeyError: 'BEiT is not in the backbone registry'

SamXiaosheng commented 2 years ago

i follow the steps using https://github.com/SamXiaosheng/unilm/tree/master/beit/semantic_segmentation, how to solve it?thanks advance.

addf400 commented 2 years ago

We use a specified version of mmseg lib(mmcv-full==1.3.0 & mmsegmentation==0.11.0) for reproduction. Please follow the introduction for installing the required packages.

SamXiaosheng commented 2 years ago

thanks for your answer, i check the version of mmseg lib(mmcv-full==1.3.0 & mmsegmentation==0.14.1) ,mmsegmentation is higher verison , is it possible the problem?

KaiChen1998 commented 2 years ago

We use a specified version of mmseg lib(mmcv-full==1.3.0 & mmsegmentation==0.11.0) for reproduction. Please follow the introduction for installing the required packages.

But this version of mmsegmentation does not support either BEiT-version of ViT or the original ViT...

addf400 commented 2 years ago

We use a specified version of mmseg lib(mmcv-full==1.3.0 & mmsegmentation==0.11.0) for reproduction. Please follow the introduction for installing the required packages.

But this version of mmsegmentation does not support either BEiT-version of ViT or the original ViT...

We follow Swin/XCiT codebases to add some code for supporting BEiT in mmseg for a specified version (mmcv-full==1.3.0 & mmsegmentation==0.11.0). More details can be found in our release code Following the setup by introduction, the code has been tested and can be executed normally.