jason71995 / Keras_ODENet

Implementation of (2018) Neural Ordinary Differential Equations on Keras
64 stars 18 forks source link

Training freezing #6

Open myagmur01 opened 5 years ago

myagmur01 commented 5 years ago

Hi Jason,

Thanks for sharing. When I try training the ode-block, seeing no response in terminal, it keeps waiting for long time and I am killing the process at the end. When trying for Resnet, it is working and getting results as normal. Do you have any suggestion about this? I am actually using gtx 1080 Ti with 12 gb and I think it should work.

jason71995 commented 5 years ago

Hi @mytrjp

Which version of tensorflow and keras you used? run on GPU or CPU?

I tried tensorflow 1.13.1 on CPU, it work normally.