diambra / arena

DIAMBRA Arena: a New Reinforcement Learning Platform for Research and Experimentation
https://docs.diambra.ai
Other
313 stars 22 forks source link

Add container images #48

Closed discordianfish closed 1 year ago

discordianfish commented 1 year ago

@alexpalms I've tested sb3 with my agent and it works. Can you give me a sb2 and raylib sample agent? Then I can test this as well.

alexpalms commented 1 year ago

@alexpalms I've tested sb3 with my agent and it works. Can you give me a sb2 and raylib sample agent? Then I can test this as well.

@discordianfish

Stable Baselines

You can use the agent we have here: https://github.com/diambra/agents/tree/main/stable_baselines

Making sure that you provide to the pre-build SB image the following (as secrets?):

This should be enough, BUT, I don't see the tensorflow installation instruction in the SB dockerfile (see previous comment in history), I think without it it won t work.

RLlib

I never trained something, my desktop computer should arrive in a few days and I will restart with my background trainings. The fastest way should be to just run a simple training using one of the examples here https://github.com/diambra/agents/tree/main/ray_rllib and simply dump the model after a single iteration. The creating an agent using it.

But I suggest the following: if you can, test SB, I will use RL lib to run my tests about how to use these pre built images wdyt?