xulianuwa / MCTformer

Code for CVPR 2022 paper "Multi-Class Token Transformer for Weakly Supervised Semantic Segmentation"
161 stars 16 forks source link

RuntimeError #43

Open geek-APTX4869 opened 11 months ago

geek-APTX4869 commented 11 months ago

Unknown model (deit_small_MCTformerV2_patch16_224)

chen-zhoujian commented 7 months ago

I encountered the same problem. There is no model implementation of "deit_small_MCTformerV2_patch16_224" in the timm library.

zbb1111 commented 7 months ago

我遇到了,同样的问题,请问你们解决了吗!?

djene-mengistu commented 6 months ago

you have to just registor a model by this name in 'model.py', just like deit_small_MCTformerPlus.

@register_model def deit_small_MCTformerV1_patch16_224(pretrained=True, **kwargs):