jiawei-ren/diffmimic

[ICLR 2023] DiffMimic: Efficient Motion Mimicking with Differentiable Physics https://arxiv.org/abs/2304.03274

diffmimic's Issues

Start my own training

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

run code with new version of brax and jax ?

Hi,

Do you think the code may run with new version of brax and jax ?

I have:
brax 0.10.5
brax-jumpy 0.1.0
flax 0.8.4
jax 0.4.29
jax-cuda12-pjrt 0.4.29
jax-cuda12-plugin 0.4.29
jaxlib 0.4.29
jaxopt 0.8.3
optax 0.2.2
orbax-checkpoint 0.5.16

I tested but I have issue with jumpy.
I can search but I'd like to know if it's worth it or if too much has changed in Brax and Jax ?

Thanks

AttributeError: module 'scipy.linalg' has no attribute 'tril'

Hello,
Thanks to your work! And when I want to run the start code as README , there is a problem, and it is difficult to solve.

data type question

Thanks for the awesome work , i have a question does this method works with mocap data only or does it work with video data too

jaxlib is version 0.3.20, but this version of jax requires version >= 0.4.7.