Engineering-Course / LIP_JPPNet

Code repository for Joint Body Parsing & Pose Estimation Network, T-PAMI 2018
MIT License
341 stars 87 forks source link

Problem when training #54

Open M-Ghorbel opened 5 years ago

M-Ghorbel commented 5 years ago

Hello, i'm trying to run train_JPPNet-s2.py but i'm getting this error:

InvalideArgumentError (see above for traceback): Cannot assign a device for operation 'Tower_0/stridded_slice_4': Could not satisfy explicit device specification '/device:GPU:0' because no kernel for GPU devices is available. Any help please? thank you