filipetrocadoferreira / end2endlobesegmentation

Code to replicate Experiments from the paper 'End-to-End Supervised Lobe Segmentation'
40 stars 12 forks source link

Train problem #6

Open Endless-Hao opened 5 years ago

Endless-Hao commented 5 years ago

When i train the model in my data. I get the error blow: totalMemory: 10.91GiB freeMemory: 10.64GiB 2019-05-15 15:47:14.250672: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1512] Adding visible gpu devices: 0 2019-05-15 15:47:14.251248: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] Device interconnect StreamExecutor with strength 1 edge matrix: 2019-05-15 15:47:14.251256: I tensorflow/core/common_runtime/gpu/gpu_device.cc:990] 0 2019-05-15 15:47:14.251260: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1003] 0: N 2019-05-15 15:47:14.251381: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 10350 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1) Traceback (most recent call last): File "/home/pch/7-med/end2endlobesegmentation-master/end2endlobesegmentation-master/train.py", line 241, in train(parser.parse_args()) File "/home/pch/7-med/end2endlobesegmentation-master/end2endlobesegmentation-master/train.py", line 235, in train callbacks=[checker, tb, stopper, saver]) File "/home/pch/anaconda3/envs/p27tkt/lib/python2.7/site-packages/keras/legacy/interfaces.py", line 91, in wrapper return func(*args, *kwargs) File "/home/pch/anaconda3/envs/p27tkt/lib/python2.7/site-packages/keras/engine/training.py", line 1418, in fit_generator initial_epoch=initial_epoch) File "/home/pch/anaconda3/envs/p27tkt/lib/python2.7/site-packages/keras/engine/training_generator.py", line 181, in fit_generator generator_output = next(output_generator) File "/home/pch/anaconda3/envs/p27tkt/lib/python2.7/site-packages/keras/utils/data_utils.py", line 709, in get six.reraise(sys.exc_info()) File "/home/pch/anaconda3/envs/p27tkt/lib/python2.7/site-packages/keras/utils/data_utils.py", line 685, in get inputs = self.queue.get(block=True).get() File "/home/pch/anaconda3/envs/p27tkt/lib/python2.7/multiprocessing/pool.py", line 572, in get raise self._value ValueError: need more than 1 value to unpack Epoch 1/4000

Process finished with exit code 1

But i don't know how to solve.What's wrong?