xmengli / H-DenseUNet

TMI 2018. H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes
527 stars 161 forks source link

about train_2ddense.py #93

Closed sipl-zrc closed 4 years ago

sipl-zrc commented 4 years ago

Hi, I'm running my own dataset, but i don't know why the code stopped iterating. The likely cause?(there is enough memory)

Epoch 1/6000 80/13693 [..............................] - ETA: 14000s - loss: 0.3437Traceback (most recent call last): File "/home/ubuntu/H-DenseUNet-master/train_2ddense.py", line 221, in train_and_predict() File "/home/ubuntu/H-DenseUNet-master/train_2ddense.py", line 217, in train_and_predict workers=3, use_multiprocessing=True) File "Keras-2.0.8/keras/legacy/interfaces.py", line 87, in wrapper return func(*args, **kwargs) File "Keras-2.0.8/keras/engine/training.py", line 2011, in fit_generator generator_output = next(output_generator) StopIteration 81/13693 [..............................] - ETA: 13874s - loss: 0.3408 Process finished with exit code 1