ValueError: could not broadcast input array from shape (2,15) into shape (2,16)

[liusui652]

When I ran train.py, this bug appeared. Later, I adjusted the torch version and no longer reported the error, but it appeared again when I ran evaluate_model.py. I don't know what caused it?

[eunjic]

@liusui652 I'm having same problem when I run train.py. What torch version did you adjusted?

[lsch0lz]

Hi @liusui652 ! Did you yolved the problem? I'm currently facing the same issue

[Nadzi77]

It is because some person is not in all 16 positions (8 observed, 8 predicted) - in your case, "(2,15)" means there are 15 positions of some person in dataset instead of 16. I had the same issue so I added these lines in trajectory.py under line curr_ped_seq = np.transpose(curr_ped_seq[:, 2:]):

if curr_ped_seq.shape != (2,16): continue

(in case you are predicting 12 positions, use (2,20) )