error in training step 2

CongHan0808 / DeOP

Open-vocabulary Semantic Segmentation

32 stars 3 forks source link

error in training step 2 #1

Closed bingnoi closed 6 months ago

bingnoi commented 8 months ago

python train_net.py --config-file configs/coco-stuff-164k-156/zero_shot_maskformer_R101c_bs32_60k.yaml --num-gpus 8 MODEL.CLIP_ADAPTER.PROMPT_CHECKPOINT ${TRAINED_PROMPTS}

File "DeOP/mask_former/modeling/criterion_decoder.py", line 211, in target["sem_seg"] for target in targets KeyError: 'sem_seg'

CongHan0808 commented 8 months ago

You can try to set "MODEL.MASK_FORMER.ONLY_MASKFORMER = True". During training MaskFormer, criterion.py is needed.

bingnoi commented 8 months ago

thanks,but got another problem. mask_former/modeling/matcher.py", line 162, in memory_efficient_forward raise ValueError("matrix contains nan in mask") ValueError: matrix contains nan in mask

where got wrong?