quancore / social-lstm

Social LSTM implementation in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TypeError occurs when running train.py

lk-greenbird opened this issue · comments

Hi,
When I excute train.py, a TypeError occurs.It looks like this,"TypeError: indexing a tensor with an object of type list.".The Traceback refers to line 312 in train.py, "outputs, _, _ = net(x_seq, grid_seq,......)."(net = SocialModel(args)). And it also refers to some torch documents.
image
My environment includes python3.6 and pytorch0.1.12 on ubuntu16.04. I want to know weather this problem dues to the version of
pytorch. If not, can you tell me your environment which can excute this project successfully. Thank you!