huang-xx/STGAT

reproduce the ADE/FDE for eth and zara1

howiechasefuture opened this issue · 1 comments

Hi,

Thank you for providing the code.
I am trying to train the models for eth and zara1 with prediction length as 12, but I can't achieve (ade/fde) reported in the paper. (eth: 0.65/1.12 and zara1: 0.34/0.69.)
Can you share the parameters for these two datasets?
Thank you.

@howiechasefuture For the ETH dataset, please check this issue. In my memory, for other datasets, the default parameters in the code should produce results very close to those in the paper. I can't access my lab computer right now, and I may upload the pre-trained model later.