Where is source for tensorflow gym environments implementation

Question

I need to implement custom tensorflow gym environment to use it with tf agents.

Is there a code on Github for "standard" gym environment? Eg cart pole

Please note this is tensorflow specific question not openAi

Do you mean something like [openai gym](https://github.com/openai/gym)? — code-lukas, Sep 05 '22 at 17:38

code-lukas · Answer 1 · 2022-09-05T17:48:17.560

0

In addition to my comment:

OpenAI Gym provides a cartpole example: Cartpole.py

If you need any additonal information regarding gym, Cartpole integration or Actor-Critics I can highly recommend the following lectures: Reinforcement Learning @ FAU

edited Sep 05 '22 at 17:48

answered Sep 05 '22 at 17:41

code-lukas

1,586
9
19

No I meant tensorflow own `suit_gym` - it's similar to open ai gym but it has all the power and parallelism of tensorflow – Boppity Bop Sep 05 '22 at 21:34
Also I need code of an actual environment class not a sample how to use it. I need to implement my own environment so its easier by looking at a working environment code – Boppity Bop Sep 05 '22 at 21:38

score -1 · Answer 2 · answered Sep 06 '22 at 14:45

According to this tutorial: https://www.tensorflow.org/agents/tutorials/2_environments_tutorial

you either create python (OpenAi like) environment and then use tensorflow wrapper to make it more efficient (covered in the tutorial)

or

create pure tensorflow environment, code example here: https://github.com/tensorflow/agents/blob/master/tf_agents/environments/tf_environment_test.py

Where is source for tensorflow gym environments implementation

2 Answers2