google-research / robopianist

[CoRL '23] Dexterous piano playing with deep reinforcement learning.
https://kzakka.com/robopianist/
Apache License 2.0
567 stars 46 forks source link

Availability of Training Codes #12

Closed Minyoung1005 closed 1 year ago

Minyoung1005 commented 1 year ago

Hi,

I love your work and would like to use this benchmark for my RL research. I want to reproduce DroQ and PPO results illustrated in Figure 4 in your paper.

Do you have any plans to release the training codes? I would also appreciate it if you could provide the pre-trained models.

kevinzakka commented 1 year ago

Hi @Minyoung1005, I'll try to get around releasing the RL code soon, hopefully within this month.

Minyoung1005 commented 1 year ago

I'm excited to hear that you're planning to release the RL code soon. I look forward to exploring the code once it's available.

kevinzakka commented 1 year ago

Hi @Minyoung1005, see https://github.com/kevinzakka/robopianist-rl for the RL training code.

Minyoung1005 commented 1 year ago

@kevinzakka Thank you! Checked the new repo :)