ValueError: Dimension 0 in both shapes must be equal, but are 1024 and 512. Shapes are [1024,256] and [512,256]. for 'Assign_12' (op: 'Assign') with input shapes: [1024,256], [512,256].

[GaoPengGit]

This error still occurs at Keras==2.2.4.

[ibtehajali67]

I solved this error by using pip install tensorflow-gpu=1.13.1. Run that lines !pip uninstall -y tensorflow !pip install tensorflow-gpu==1.13.1