Updated Version of Code

[RichardUVA]

Hi Do you have a updated version of the code?

Since Tensorflow 1.0 is too old and I am suffering from a layer building problem.

Thanks,
Junyu