Giters

megvii-research / MSPN

Multi-Stage Pose Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

An OOM problem in training with 1080Ti

czp97 opened this issue · comments

commented

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?