xmengli / H-DenseUNet

TMI 2018. H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

about train_2ddense.py

sipl-zrc opened this issue · comments

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