dongwu92 / AutoPortraitMatting

Tensorflow implementation of Automatic Portrait Matting on paper "Automatic Portrait Segmentation for Image Stylization"
Apache License 2.0
478 stars 151 forks source link

Crash on Titan X Pascal GPU #18

Open saeed68gm opened 7 years ago

saeed68gm commented 7 years ago

My system crashes when I run the script to train the model. I am using a Titan X Pascal GPU and even on the script is getting stuck.

I have changed the following parts of the script to accommodate updates on the new versions of tensorflow. (historgram_summary, scalar_sammary, sparse_softmax_cross_entropy_with_logits API's were giving me errors).

Anybody face the same issue?