YOLOX icon indicating copy to clipboard operation
YOLOX copied to clipboard

Does CUDA out of memory because of simOTA

Open leetesua opened this issue 4 years ago • 6 comments

I am having Cuda out of memory error. It always happens after several epoch of training. Is it possible that simOTA is the reason? Because to my understanding, after several epoch of training, the model gets better at detecting, the dynamic_k gets larger too, so it means more positive samples, to be loaded into memory. Hence cause out of memory error. am I right?

leetesua avatar Oct 19 '21 07:10 leetesua

@leetesua are you using bigger dataset than COCO ?

ajtvsv07 avatar Oct 25 '21 12:10 ajtvsv07

In yolox training logic, image size is changed per 10 iters, which causes the gpu memory raising/falling.

FateScript avatar Oct 29 '21 05:10 FateScript

@leetesua are you using bigger dataset than COCO ?

my dataset is much smaller than COCO,

leetesua avatar Nov 03 '21 02:11 leetesua

In yolox training logic, image size is changed per 10 iters, which causes the gpu memory raising/falling. are you referring to dynamic_scale in the train dataset config file?

In yolox training logic, image size is changed per 10 iters, which causes the gpu memory raising/falling.

where Can I change this setting?

leetesua avatar Nov 03 '21 03:11 leetesua

Your exp file.

FateScript avatar Nov 03 '21 04:11 FateScript

In yolox training logic, image size is changed per 10 iters, which causes the gpu memory raising/falling. are you referring to dynamic_scale in the train dataset config file?

In yolox training logic, image size is changed per 10 iters, which causes the gpu memory raising/falling.

where Can I change this setting?

For people concerned, change random_size or multiscale_range in exp file.

Jackie-LJQ avatar Apr 22 '24 05:04 Jackie-LJQ