search

Denys88 / rl_games

RL implementations
MIT License
848 stars 142 forks source link

Add wandb integration #167

Closed vwxyzjn closed 2 years ago

vwxyzjn commented 2 years ago

Feel free to run 

python runner.py --train --file rl_games/configs/atari/ppo_breakout_torch.yaml --track
WANDB_API_KEY=xxxx python runner.py --train --file rl_games/configs/atari/ppo_breakout_torch.yaml --track
python runner.py --train --file rl_games/configs/atari/ppo_breakout_torch.yaml --wandb-project-name rl-games-special-test --track
python runner.py --train --file rl_games/configs/atari/ppo_breakout_torch.yaml --wandb-project-name rl-games-special-test -wandb-entity openrlbenchmark --track
Denys88 commented 2 years ago

Thanks! Can I merge it?

vwxyzjn commented 2 years ago

I’d suggest merging poetry first, then I can add the dependencies in this PR and documentation :). Otherwise feel free to merge it.

Denys88 commented 2 years ago

Just merged poetry

vwxyzjn commented 2 years ago

Give me a sec.

vwxyzjn commented 2 years ago

It's ready. Thanks!