xulianuwa / MCTformer

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

RuntimeError #43

Open geek-APTX4869 opened 5 months ago

geek-APTX4869 commented 5 months ago

Unknown model (deit_small_MCTformerV2_patch16_224)

chen-zhoujian commented 2 months ago

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

zbb1111 commented 1 month ago

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

djene-mengistu commented 1 month 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):