KiseKloset / DM-VTON

👗 DM-VTON: Distilled Mobile Real-time Virtual Try-On
https://sites.google.com/view/ltnghia/research/DMVTON
Other
107 stars 23 forks source link

Pre-trained pb checkpoints #21

Closed serdarnazli closed 3 months ago

serdarnazli commented 4 months ago

Is it possible for us to obtain pre-trained pb checkpoints? Training the teacher model from scratch is time consuming making further research less viable

rusenbb commented 4 months ago

Hey maintainers! 🙌

I couldn't agree more with the original poster. Training the Teacher model from scratch is a huge time sink, and it's holding back a lot of us who are itching to dive into the exciting parts of DM-VTON research.

It would make the project way more accessible and allow us to focus on pushing the boundaries of what's possible. I know it's extra work, but I genuinely believe it would supercharge innovation and collaboration in the community.

So, pretty please, could you consider sharing those pre-trained pb checkpoints? It would mean the world to us! 🙏

Thanks for all your hard work and for considering this request. You're awesome! 😄

Cheers

zero-nnkn commented 3 months ago

Thank you for your interest in the research. I created a release 1.0.0 with the teacher/student pretrained checkpoints that were reported in the paper. I hope it will be useful.