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

MAE Pre-Training Weights #7

Closed a-pru closed 8 months ago

a-pru commented 8 months ago

Hi,

Could you possible provide the checkpoint of the MAE model after pre-training? Really appreciate your work!

Thanks.

Best, Alex

jchengai commented 8 months ago

Hi @a-pru, the mae ckpt is updated to the README along with a demo notebook.