How much memery is needed when train with single GPU?

lzccccc / SMOKE

SMOKE: Single-Stage Monocular 3D Object Detection via Keypoint Estimation

MIT License

707 stars 177 forks source link

How much memery is needed when train with single GPU? #29

Open lidehuihxjz opened 4 years ago

lidehuihxjz commented 4 years ago

I trained the model wiht single GPU when batchsize is 2 and got error of out of memery. I tried to test with a very simple backbone, containing only 4 downsample layers, but also got the same error. My total GPU memery is 8G.

YuchenWu61 commented 4 years ago

11gb