Downloads · 30 days
0
mario-rl-model
akantox/mario-rl-model
mario-rl-model is a reinforcement learning model from akantox. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This is a PPO agent trained using Stable Baselines3 and Gymnasium on a Mario-like environment.
Access
Public
Updated Apr 6, 2025
Repo size
1.7 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.zip580 MB · 100%
At a glance
- Task
- Reinforcement Learning
- License
- mit
- Access
- Public
- Created
- Mar 30, 2025
- Updated
- Apr 6, 2025
- SHA
- 0e76e665
Model card
From the Hugging Face model README
Mario PPO Model
This is a PPO agent trained using Stable Baselines3 and Gymnasium on a Mario-like environment.
Environment Details
- Action Space: Simple discrete NES-style actions (7 total)
- Observation: Grayscale, 250×264
- Frame Stack: 4 frames
Training Info
- Algorithm: PPO
- Framework: Stable Baselines3
- Timesteps: 20 million
- Environment: Gymnasium (
v0) - Device: MPS / CUDA / CPU
Training Timesteps & Checkpoints
Usage
from stable_baselines3 import PPO
from huggingface_hub import hf_hub_download
model_path = hf_hub_download(repo_id="akantox/mario-rl-model", filename="mario_ppo.zip")
model = PPO.load(model_path)
- Task
- Reinforcement Learning
- License
- mit
- Languages
- en
- Created
- Mar 30, 2025
- Updated
- Apr 6, 2025