checkpoints of pre-training

ma-xu / Rewrite-the-Stars

[CVPR 2024] Rewrite the Stars

Apache License 2.0

216 stars 14 forks source link

checkpoints of pre-training #4

Closed GrimaceLittleBoss closed 1 month ago

GrimaceLittleBoss commented 2 months ago

Thank you for your excellent work! Could you please provide the checkpoints of pre-training?

ma-xu commented 2 months ago

@GrimaceLittleBoss Sorry for late reply from Github. Besides the s4 checkpoints I sent via email, I uploaded all checkpoints and logs for StarNet and included in the model python file. You can directly load via "pretrained=True".

Thanks for your interest, please feel free to let me know if any other questions or concerns.

Best, Xu