MCG-NJU/MixFormer

Can we run it on windows wiht torch 2.0.1 ?

Opened this issue · 1 comments

Thanks for open-sourcing your project but can we run it on windows with pytorch 2.0.1 as the prroi_pooling dependency seams to conflicting for windows ?

I recommend you replace prroi_pooling with torchvision.ops.roi_align. Maybe you should re-train the score prediction module (SPM) and it does not cost too much time.