MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
http://mvig.org/research/alphapose.html
Other
7.96k stars 1.97k forks source link

Epoch is not running when trying to train alphapose project #974

Closed dekathomas closed 2 years ago

dekathomas commented 2 years ago

Hello, I try to train the model in Google Colab, but I never got the epoch information there, why this can happen? image The red line is the log I made

I have try to train it in my local PC and it works, but my PC doesn't have enough powerful GPU so I have to use Google Colab to run my training image

Can you help me to solve this problem? @Fang-Haoshu

Thank you.

Fang-Haoshu commented 2 years ago

Hi, sorry that I am not sure what happened with the colab. Will it give any errors if you press ctrl+c on the Colab?

dekathomas commented 2 years ago

Hi @Fang-Haoshu it doesn't give any errors to me. Have you ever tried to train AlphaPose using colab?

dekathomas commented 2 years ago

hello @Fang-Haoshu this is what I got when I stop it image

dekathomas commented 2 years ago

@Fang-Haoshu do you think it's because alphapose take so much resource of memory to load the dataset? Because it's stuck when it tries to load the dataset