Open DJC666666 opened 1 month ago
I will look into this issue, for now, you can change the merge_type to hierarchical, here: https://github.com/hovsg/HOV-SG/blob/6e7560061ec5ea1f11670e8ebcbda255818b8f24/config/create_graph_config.yaml#L35 this may solve the problem.
Thank you for the solution, but it still does not work.
[2024-07-24 13:54:24,706][root][INFO] - Loaded ViT-H-14 model config.
[2024-07-24 13:54:30,576][root][INFO] - Loading pretrained ViT-H-14 weights (checkpoints/laion2b_s32b_b79k.bin).
Creating RGB-D point cloud: 100%|█████████████| 226/226 [00:15<00:00, 14.63it/s]
Extracting features: 0%| | 0/226 [00:00<?, ?it/s]
CUDA out of memory: CUDA out of memory. Tried to allocate 1024.00 MiB. GPU
When I tried to create scene graphs, it showed 'CUDA out of memory'. Is there any good solution?