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

Segmentation fault (core dumped) #17

Closed manthan-dl closed 11 months ago

manthan-dl commented 11 months ago

Hi Team,

Thank you for the code and pre-trained model. But when I am trying to load model to GPU memory I am getting "Segmentation fault (core dumped)" message. I am currently using 16 GB GPU machine with the same environment as the repository suggested. Also tried with 24 GB machine but still getting the same error. What could be the reason for this? And if anyone has any solution kindly suggest the same.

  1. Is the DETR implementation only works on v100 GPUs?
  2. Do I need single GPU with almost 32 GB of memory?
  3. Is there a way to implement the same with CPU?