salinasJJ / BBpose

Human Pose Estimation using Soft-gated Skip Connection and base Hourglass Networks. Implemented in Tensorflow 2.
MIT License
29 stars 11 forks source link

GPU not activated #8

Open xavi0007 opened 8 months ago

xavi0007 commented 8 months ago

Was able to get the code to start training on A100 cluster but it's not utilising the GPU at all.