Reinforcement Learning
stable-baselines3
4d-snake
4d-snake-2x4
snake
deep-reinforcement-learning
sb3-contrib
maskable-ppo
Eval Results (legacy)
Instructions to use BurnyCoder/4d-snake-exp02-ppo-2x4 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- stable-baselines3
How to use BurnyCoder/4d-snake-exp02-ppo-2x4 with stable-baselines3:
from huggingface_sb3 import load_from_hub checkpoint = load_from_hub( repo_id="BurnyCoder/4d-snake-exp02-ppo-2x4", filename="{MODEL FILENAME}.zip", ) - Notebooks
- Google Colab
- Kaggle
File size: 932 Bytes
929734b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | {
"size": 2,
"ndim": 4,
"idle_mult": 4,
"r_food": 1.0,
"r_death": -1.0,
"r_win": 10.0,
"r_step": -0.001,
"shaping_coef": 0.0,
"n_envs": 1024,
"total_timesteps": 5000000,
"n_steps": 32,
"batch_size": 4096,
"n_epochs": 4,
"gamma": 0.99,
"gae_lambda": 0.95,
"lr_start": 0.0003,
"lr_end": 1e-05,
"clip_start": 0.2,
"clip_end": 0.05,
"ent_coef": 0.01,
"vf_coef": 0.5,
"max_grad_norm": 0.5,
"target_kl": 0.03,
"net_width": 512,
"device": "auto",
"torch_threads": 8,
"seed": 0,
"curriculum": 0,
"curriculum_window": 8,
"curriculum_delta": 0,
"curriculum_rho": 0.2,
"curriculum_min_eps": 200,
"p_true_start": 0.2,
"eval_episodes": 100,
"eval_every": 327680,
"ckpt_every": 1310720,
"eval_seeds": "0,1,2",
"bench_steps": 200000,
"runs_dir": "runs",
"run_name": "exp02_ppo_2x4",
"model_path": "",
"policy": "route"
} |