jiawei-ren / diffmimic

[ICLR 2023] DiffMimic: Efficient Motion Mimicking with Differentiable Physics https://arxiv.org/abs/2304.03274
https://diffmimic.github.io/
Other
278 stars 21 forks source link

Start my own training #4

Closed yinkangning0124 closed 1 year ago

yinkangning0124 commented 1 year ago

Hi jiawei, Thanks for your awesome work. I want to know how to train the model with my own amass dataset, can you give me a detailed README on how to start training? Another question is that is there any pre-trained model? Best, Kangning

jiawei-ren commented 1 year ago

Hi Kangning,

You can try the converter script here, it works with the official AMASS dataset.

The request for pre-trained models is noted, I will upload them when I get the time. But generally training a policy should take less than two hours.