COde giving cocate error for other dataset

[swapna-2205]

I tried to run the code for other datasets generated using MIMO Data generator.
I generated for 16 antennas and 8 users, but the python main.py code showing the error
InvalidArgumentError: ConcatOp : Dimensions of inputs should match: shape[0] = [1,2,2,512] vs. shape[1] = [1,1,2,512] [Op:ConcatV2] name: concat

It is not working for any other dataset. please help me this regard