search

dongbo811 / AFFormer

119 stars 12 forks source link

KeyError: "EncoderDecoder: 'afformer_base is not in the models registry'" #3

Closed SAWRJJ closed 1 year ago

SAWRJJ commented 1 year ago

Hello, thank you for the excellent paper, I also try to reproduce the model in the paper, but when I load the model it will appear KeyError: "EncoderDecoder: 'afformer_base is not in the models registry'". I hope you can answer my questions in your busy schedule

liyaoshun commented 1 year ago

Hello, thank you for the excellent paper, I also try to reproduce the model in the paper, but when I load the model it will appear KeyError: "EncoderDecoder: 'afformer_base is not in the models registry'". I hope you can answer my questions in your busy schedule modify /AFFormer/mmseg/models/backbones/init.py && copy /AFFormer/tools/afformer.py to */AFFormer/mmseg/models/backbones/afformer.py

from .resnet import ResNet from .afformer import afformer_base from .afformer import afformer_small from .afformer import afformer_tiny

all = [ 'ResNet', 'afformer_base', 'afformer_small', 'afformer_tiny' ]

SAWRJJ commented 1 year ago

Hello, thank you for the excellent paper, I also try to reproduce the model in the paper, but when I load the model it will appear KeyError: "EncoderDecoder: 'afformer_base is not in the models registry'". I hope you can answer my questions in your busy schedule modify /AFFormer/mmseg/models/backbones/init.py && copy /AFFormer/tools/afformer.py to */AFFormer/mmseg/models/backbones/afformer.py

from .resnet import ResNet from .afformer import afformer_base from .afformer import afformer_small from .afformer import afformer_tiny

all = [ 'ResNet', 'afformer_base', 'afformer_small', 'afformer_tiny' ]

ty

dongbo811 commented 1 year ago

Thank you for your attention!