Can the project run on GPU?

[Wisgon]

I use tensorflow-gpu to run the code, but it still very slow and the gpu-util is always 0, and the memory of my two GPUs both full.

[Wisgon]

@SaoYan

[Yiqi Yan]

I ran the code on GPU.

It is true that this code is memory consuming. More improvements are encouraged.

[Wisgon]

OK, thank you for replying.