Open SouthMemory opened 7 years ago
I had trained the models using the theano backend, so not sure about the TensorFlow backend. That said, I hope you changed line 22 of LUNA_unet.py K.set_image_dim_ordering('th') # Theano dimension ordering in this code To K.set_image_dim_ordering('tf') # Tensorflow dimension ordering.
Hello,Thank you very much. I tried your advice, but it didn't work , some else error occured But one thing you are right is that the problem is about the backend So I tried to change my backend from tensorflow to theano, by editing /home/beichuan/.keras/keras.json, and then your code worked.
But I found that on my computer your code runs on CPU and not on GPUs, so it is quite slow for me. Did you run it on CPU or GPU? Did I miss something?
Thank you!
On a K40 GPU. Make sure your theano configurations are correct. See Install and configure GPU drivers http://deeplearning.net/software/theano/install_ubuntu.html
hello, I am facing some error while installing theano for gpu as follow, RuntimeError: To use MKL 2018 with Theano you MUST set "MKL_THREADING_LAYER=GNU" in your environement. how should I solve this?? Thanks in advance!
I got some error when i run the code by using theano backend as follow,
RuntimeError: error getting worksize: CUDNN_STATUS_BAD_PARAM
Apply node that caused the error: GpuDnnConv{algo='small', inplace=True, num_groups=1}(GpuContiguous.0, GpuContiguous.0, GpuAllocEmpty{dtype='float32', context_name=None}.0, GpuDnnConvDesc{border_mode='half', subsample=(1, 1), dilation=(1, 1), conv_mode='conv', precision='float32', num_groups=1}.0, Constant{1.0}, Constant{0.0})
Toposort index: 421
Inputs types: [GpuArrayType
Backtrace when the node is created(use Theano flag traceback.limit=N to make it longer):
File "/home/coe/anaconda2/lib/python2.7/site-packages/spyder/utils/site/sitecustomize.py", line 705, in runfile
execfile(filename, namespace)
File "/home/coe/anaconda2/lib/python2.7/site-packages/spyder/utils/site/sitecustomize.py", line 94, in execfile
builtins.execfile(filename, *where)
File "/home/coe/UNET_CODE FROM GITHUB/Luna2016-Lung-Nodule-Detection-master/UNET/Code/LUNA_unet.py", line 219, in
HINT: Use the Theano flag 'exception_verbosity=high' for a debugprint and storage map footprint of this apply node. Thanks in advance!!
I got into some trouble when trying to train the model:
TypeError Traceback (most recent call last)