lucidrains / q-transformer

Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind
MIT License
304 stars 15 forks source link

integrate into stable-baselines3 #5

Open wenjun90 opened 7 months ago

wenjun90 commented 7 months ago

Hi @lucidrains, Thank you for your wonderful job. May I ask you how to integrate into stable-baselines3, please?

Best regards

lucidrains commented 7 months ago

ah hey, thanks for the kind words.

the repository isn't where I'd like it to be yet. probably by middle of the month I can start looking into that?

lucidrains commented 7 months ago

also, in the spirit of open source, always welcome any robotics / RL phd students to help with reviewing the code, pull requests, etc

wenjun90 commented 7 months ago

ah hey, thanks for the kind words.

the repository isn't where I'd like it to be yet. probably by middle of the month I can start looking into that?

What a great idea and I can't wait for this. I hope to be able to review code with you for the AI community