search

IntelLabs / coach

Reinforcement Learning Coach by Intel AI Lab enables easy experimentation with state of the art Reinforcement Learning algorithms
https://intellabs.github.io/coach/
Apache License 2.0
2.32k stars 460 forks source link

Tf2 migration #430

Open dandanelbaz opened 4 years ago

AustinDeric commented 4 years ago

What is the current working status of this? Can i start using this branch for training? (Also, thanks for the great work! :+1: ) If there is anything i can do to help, please let me know.

I noticed that the setup.py in this PR still has tensorflow >=1.9: https://github.com/NervanaSystems/coach/blob/70b7ad4c83862543893893a68e837f70d402d598/setup.py#L77

dandanelbaz commented 4 years ago

Hi Austin. This branch implements two algorithms with TF2, DQN and PPO. It should run with TF2.