dimension error

[rajeswar18]

Running the code throws a run-time error. Any help on how to resolve this, please?

Traceback (most recent call last):
File "main.py", line 311, in
prediction = onestep_models[mdl](input_state, input_action)
File "/mnt/home/plan2explore-pytorch/models.py", line 96, in call
x = torch.cat([prev_state, prev_action], dim=-1) #torch.Size([49, 50, 30]) torch.Size([49, 50, 6])
RuntimeError: Sizes of tensors must match except in dimension 0. Got 50 and 49