Logic of the index "3" in batch of evaluate visualization

[abhi-rf]

Hi,

I am trying to figure out the logic of using the index value 3 in gt=pred_traj_gt[:,3,:].data https://github.com/marsmarcin/Visual-of-Social-GAN/blob/master/plot_model03.py#L180 and the lines following it. It seems that it fails to work for batch size = 1 in the method evaluate() in model_03.py.

Thanks A lot!