Skip to content

OpenAI baselines in pytorch #12

@deontologician

Description

@deontologician

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

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions