Invalid argument: ConcatOp : Dimensions of inputs should match: shape[0] = [1,2268,81] vs. shape[2] = [1,4000,8]

[wwdok]

Hello, @markshih91 , thanks for your repo first ! I cloneed this repo, and try to adapt it to TF 2.1 -gpu by modifying a few API. After i modified all the APIs, and run inference.py, pycharm shows below error: Could you give me some suggestion to fix it ? thanks！ 哈喽，博主你好，首先感谢你上传了这个仓库 我下载了这个仓库，并试图把它改编成适应TF2.1-gpu，这主要就是改改API 当我把这些API都改完之后再运行inference.py，pycharm报出了上图的错误 你能给一些建议来解决它吗？谢谢！