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

Tensorflow training stops in iter 6147 #28

Open csf0429 opened 6 years ago

csf0429 commented 6 years ago

Tensorflow training always stops When training iterations reaches 6147. It seems the loss stills very high, So what should I do to keep model keeps training until convergence?

GateStainer commented 6 years ago

I think the loss it reports is just the loss over one batch data set, not the whole data set. So it seems to be high. Computing loss over whole training data may be too costly. The author just trains the model once over whole training set. I suggest to randomly put part of the training data into a set to compute loss?

engrchrishenry commented 6 years ago

@csf0429 I am facing the same problem. While running FCN.py the training stops at 6100 step. Did you solve it? screenshot_1

csf0429 commented 6 years ago

@engrchrishenry The author just trains the model once over whole training set. The batch data is provided by training list. You can change the condition in the training loop and set epochs to keep its training.