An error occurred during training

ZwwWayne / K-Net

[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation

Apache License 2.0

467 stars 45 forks source link

Open MiladSoleymani opened 2 years ago

MiladSoleymani commented 2 years ago

Hi, I want to train K-net on a dataset that contains 26 labels, and after running the command below,

PYTHONPATH='.':$PYTHONPATH mim train mmseg $CONFIG $WORK_DIR

I get this error

TypeError: IterativeDecodeHead: KernelUpdateHead: __init__() got an unexpected keyword argument 'mask_upsample_stride'

thanks for your help 🙏🙏