Change code to use in the newer version of TensorFlow

[ghost]

I'm using TensorFlow of version 1.10.0. I need to do the following change in file main.py to run this code:

1. Change size() into size on line 99 & 104;
2. Add tf.local_variables_initializer() after line 247&330 and add sess.run() after line 251&334.

Add I created a pull request for this. I also made some changes on the code style.