cuda out of memory about eva HOT 5 CLOSED

ps：my memory size is 32g，when i change the input into 640*640，it still happens.

from eva.

you need to change the batch size = 1
because the default image size is 1280
so if you train with batch size = 1 with image size 1280 you need at least more than 32 GB of GPU like tesla A100

for me i train on google colab pro but the max GPU i can get is A100 so my max batch size need to be 1 if i want to use eva

so you need to change the batch size in
baaivision/EVA/tree/master/det/projects/ViTDet/configs/COCO)/cascade_mask_rcnn_vitdet_eva.py

from eva.

you need to change the batch size = 1 because the default image size is 1280 so if you train with batch size = 1 with image size 1280 you need at least more than 32 GB of GPU like tesla A100

for me i train on google colab pro but the max GPU i can get is A100 so my max batch size need to be 1 if i want to use eva

so you need to change the batch size in baaivision/EVA/tree/master/det/projects/ViTDet/configs/COCO)/cascade_mask_rcnn_vitdet_eva.py

thanks for your reply. but there are 8 gpus,so my total batch size is 8(must be divided by 8)

from eva.

What is the minimum GPU size requirement?

from eva.

What is the minimum GPU size requirement?

We use A100-40G, please refer to our paper for details.

from eva.