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