search

ywyue / RoomFormer

[CVPR 2023] RoomFormer: Two-level Queries for Single-stage Floorplan Reconstruction
https://ywyue.github.io/RoomFormer/
MIT License
163 stars 22 forks source link

MSDeformAttnFunction.apply, can be implemented by pytorch? #9

Closed zuixiaosanlang closed 1 year ago

zuixiaosanlang commented 1 year ago

i want to save the model by jit.trace: model = torch.jit.trace() model.save(xxx.pt)

but failed image

thanks.

ywyue commented 1 year ago

Hi, we just use the MSDeformAttn from Deformable-DETR. They implement the MSDeformAttn using C.