ruohuali/SocialLSTM

some problems about train

Opened this issue · 0 comments

Hi, thanks for you contribution on social-lstm. when I run the code:
python3 main.py "v" --dataset eth --pure_val_name ./trained_models/s_eth.pt --T_obs 8 --T_pred 12
i got the following error:
image
I found that the problem may be caused by a mismatch between the dimensions of coords[traj_idx] and ltcorner in the model.py file, but I don't know how to fix this. This work is very important to me, could you help me. If it is convenient, here is my email 627193950@qq.com.