Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

OpenAI baselines in pytorch #12

Open
deontologician opened this issue Dec 17, 2017 · 7 comments
Open

OpenAI baselines in pytorch #12

deontologician opened this issue Dec 17, 2017 · 7 comments

Comments

@deontologician
Copy link
Contributor

There are several nice implementations of RL algorithms in

https://github.com/openai/baselines

But they're all implemented in Tensorflow, and in addition have lots of onerous dependencies like MPI. It would be nice to have these implementations in pytorch, and using something other than MPI for communication (maybe [gRPC][(https://github.com/grpc/grpc) ? )

Someone has started a reimplementation of the baselines in pytorch here:

https://github.com/nadavbh12/baselines-pytorch

but they've only implemented A2C so far. It would be nice if we also had an implementation of ACKTR and PPO2

@shicode
Copy link
Collaborator

shicode commented Dec 18, 2017

I can take a shot at this! Github username: shicode

@deontologician
Copy link
Contributor Author

Cool, I added you to the project. What I'd suggest is cloning the repo, creating a new branch, and opening a pull request with what you're working on. We can talk in gitter if you want to discuss the process in more detail

@deontologician
Copy link
Contributor Author

I'm thinking maybe attempt ACKTR first. The original paper is here:

https://arxiv.org/abs/1708.05144

And the OpenAI code for it is here:

https://github.com/openai/baselines/tree/master/baselines/acktr

@shicode
Copy link
Collaborator

shicode commented Dec 19, 2017 via email

@deontologician
Copy link
Contributor Author

deontologician commented Dec 21, 2017 via email

@JACKHAHA363
Copy link

Is the project going on?

@deontologician
Copy link
Contributor Author

No, not really

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants