search

kienduynguyen / BoxeR

Code release for "BoxeR: Box-Attention for 2D and 3D Transformers"
MIT License
138 stars 22 forks source link

CUDA out of memory. #4

Closed lymdove closed 2 years ago

lymdove commented 2 years ago

I have built this project successfully. But when I run the test programs, it reported the lack of memory. It means that the project cannont run on my laptop? Screenshot from 2022-04-06 13-58-15

kienduynguyen commented 2 years ago

Hi. It should be ok. It yields out-of-mem error because I test the box-attention with a high dimension tensor (d=1024). You can run the training script to check further.