jchengai / forecast-mae

[ICCV'2023] Forecast-MAE: Self-supervised Pre-training for Motion Forecasting with Masked Autoencoders
https://arxiv.org/pdf/2308.09882.pdf
154 stars 16 forks source link

About the length of data #14

Closed kelen1 closed 5 months ago

kelen1 commented 5 months ago

Hi! Thanks for your wonderful work first! When I try to reproduce your code, I find that in the preprocessed data, x takes the data of the first 50 time steps, and y is after 50, but many of the original data are 0-49 in length, so y is empty, does the true value of y still exist?

jchengai commented 5 months ago

Hi, @kelen1, the regression mask is used to prevent the problem you mentioned during the training

https://github.com/jchengai/forecast-mae/blob/cb86ea92601d23a8af7713389e9fb78d7e546a65/src/model/trainer_forecast.py#L89-L91

kelen1 commented 5 months ago

Thanks!