search

SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023
https://praeclarumjj3.github.io/oneformer
MIT License
1.39k stars 129 forks source link

natten error #117

Open shippingwang opened 1 month ago

shippingwang commented 1 month ago

when I tested demo 150_16_dinat_l_oneformer_coco_100ep.pth with related config configs/coco/dinat/oneformer_dinat_large_bs16_100ep.yaml I got a natten error Input axes must be greater than or equal to the product of kernel size and dilation. Got kernel size 11, dilation 20, but dimension size was 200 without any modification

cuda 11.7 torch version 2.0.0 natten version 0.17.1

joshua-stewart-rapp commented 1 month ago

I am getting the same error. Any help would be greatly appreciated.

cuda 12.2 torch version 2.2.2 natten version 0.17.1

praeclarumjj3 commented 1 month ago

Please try with natten==0.14.6. I'll update the requirements.txt in the next commit. I believe the natten v0.17.1 version may not be compatible.

You can find the older versions here: https://www.shi-labs.com/natten/

geek-APTX4869 commented 3 weeks ago

ERROR: Could not build wheels for natten, which is required to install pyproject.toml-based projects