An OOM problem in training with 1080Ti

megvii-research / MSPN

Multi-Stage Pose Network

334 stars 63 forks source link

An OOM problem in training with 1080Ti #37

Closed czp97 closed 3 years ago

czp97 commented 3 years ago

I try to test this code in Nvidia 1080Ti with mini-batch size 32 as mentioned in paper. But an OOM problem appears in first iteration training. Are you decreasing the batch size, or do some changes?