search

MeioJane / CHR

SIXray : A Large-scale Security Inspection X-ray Benchmark in CVPR 2019
80 stars 37 forks source link

run out of memory #8

Open xiying-wei opened 4 years ago

xiying-wei commented 4 years ago

When I run the program,Instead of using the GPU, the memory utilization keeps increasing until it is 100%.How to solve this problem?Thank you.

MeioJane commented 4 years ago

When I run the program,Instead of using the GPU, the memory utilization keeps increasing until it is 100%.How to solve this problem?Thank you.

I train or test using the GPU, I do not have a try without GPU

Eli-Goldberg commented 1 year ago

I just lowered the batch size to around 8 and then it used around 4GB of VRAM. On a 16GB card you can up that number to as high as 80 (but for some reason training will not be quicker).