search

rishizek / tensorflow-deeplab-v3-plus

DeepLabv3+ built in TensorFlow
MIT License
833 stars 307 forks source link

AssertionError: batch size 3 should be greater or equal than number of images to save 8 #42

Closed GeneralLi95 closed 5 years ago

GeneralLi95 commented 5 years ago

Hello,

I'm use this repo on my own dataset. It runs well at first. However, after several steps, an error occurs

batch size 3 should be greater or equal than number of images to save 8

If I restart the 'train.py', it can run some more steps. But the error occurs again. Is there anybody have the same problem?

GeneralLi95 commented 5 years ago

fixed just reduce the batch_size