search

hustvl / WeakTr

WeakTr: Exploring Plain Vision Transformer for Weakly-supervised Semantic Segmentation
MIT License
122 stars 2 forks source link

what backbone can i select in onlineRetraining? #7

Closed massica17 closed 1 year ago

massica17 commented 1 year ago

when i set $your_model_name as vit_base_patch8_384 I got a errot FileNotFoundError: [Errno 2] No such file or directory: '$TORCH_HOME/hub/checkpoints/vit_base_patch8_384.pth' when i set 'deit' KeyError: 'deit' when i set '' Error: Got unexpected extra argument (mask_transformer) so what model_name can i select?

Yingyue-L commented 1 year ago

We suppport the models provide in OnlineRetraining/segm/config.yml, only for the vit_base_patch8_384 model you need download the checkpoint from here and move it to $TORCH_HOME/hub/checkpoints/vit_base_patch8_384.pth.