search

supernotman / RetinaFace_Pytorch

Reimplement RetinaFace with Pytorch
305 stars 68 forks source link

Out of memory #1

Closed xsacha closed 5 years ago

xsacha commented 5 years ago

How much memory do you estimate this project needs? I'm using a Titan V with 12GB and this goes out of memory with a batch size of 16 (default was 32), which seems quite small for WIDER face.

I had to use a batch size of 8, which used 10GB.

supernotman commented 5 years ago

How much memory do you estimate this project needs? I'm using a Titan V with 12GB and this goes out of memory with a batch size of 16 (default was 32), which seems quite small for WIDER face.

I had to use a batch size of 8, which used 10GB.

For me , I am using batch size of 32 and 36GB memory need. And i have not tried batch size of 8 or smaller.

xsacha commented 5 years ago

Thanks