How to run demo on GPU?

[rdverse]

Thank you for your work. This repo has helped me to fuse the keypoints and segmentation tasks together. However, when I run this it is strictly running on cpu. Is there any way I could run this model on GPU?

[rjjakes]

Read the code: https://github.com/chrispolo/Keypoints-of-humanpose-with-Mask-R-CNN/blob/master/config.py

# NUMBER OF GPUs to use. For CPU training, use 1    

GPU_COUNT = 1