Question about model output and error in graph.

[Joevaen]

Hi,
When I download your code and run train.py., an error below ocurrs.

and the output shapes are all (?, x, x) like:

I am not familiar with tensorflow1.8.0, could you please tell how to fix it?
Thanks!