How to decrease Batch size

chenyilun95 / tf-cpn

Cascaded Pyramid Network for Multi-Person Pose Estimation (CVPR 2018)

MIT License

792 stars 197 forks source link

How to decrease Batch size #91

Open lisa676 opened 4 years ago

lisa676 commented 4 years ago

Hi, how we can decrease batch size for small GPU? I tried in mpest and confiq.py file but I got error "List out of index".

benMOOVENCY commented 3 years ago

Hi lan786, did you find a way to decrease the batch size

lisa676 commented 3 years ago

@benMOOVENCY Sorry for late reply. I didn't get any idea to decrease the batch size. I'm using 11GB card and it solve my problem

benMOOVENCY commented 3 years ago

@lan786 thanks for the reply, and witch card are you using, we try to set up system to be able to retrain the model. How many time take the train of the model with your card ?

lisa676 commented 3 years ago

@benMOOVENCY Any 11GB card (i.e. 1080ti or 2080ti should be enough). I didn't use this repo for training I just used pre-trained weights.