megvii-research / MSPN

Multi-Stage Pose Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

loss is vibrating during training with one gpu

cbyzju opened this issue · comments

hi,
The loss keeps vibrating during training with one gpu, do you have any suggestions to solve the problem? training dataset is coco, thx!

You can enlarge the batch size or extend the warm-up.

how long it takes to train coco data with one gpu ?

Maybe a week? You can adjust training epochs to make it faster.