AutoPortraitMatting
AutoPortraitMatting copied to clipboard
Crash on Titan X Pascal GPU
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?