train_PNet.py doesn't start?

AITTSMD / MTCNN-Tensorflow

Reproduce MTCNN using Tensorflow

1.51k stars 710 forks source link

train_PNet.py doesn't start? #267

Open KamalM8 opened 5 years ago

KamalM8 commented 5 years ago

Hello,

I have setup my environment as was mentioned in this repository (tensorflow 1.2.1, Cuda 8.0, etc.) However, after the data preparation step, the train_PNet.py code doesn't train. It just stops at this line:

image_batch_array, label_batch_array, bbox_batch_array,landmark_batch_array = sess.run([image_batch, label_batch, bbox_batch,landmark_batch])

If anyone stumbled across this error and was able to fix it, please let me know.

Shayebh commented 5 years ago

I had the same problem.

wanglaotou commented 5 years ago

my environment is (tensorflow 1.14.0, Cuda 9.0), and it's training well.