Downloads · 30 days
1
3% of all-time downloads
Shreyboiii/dqn-pong
dqn-pong is a machine learning model from Shreyboiii. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for stable-baselines3. The card lists the license as mit.
This is a Deep Q-Learning (DQN) agent trained to play PongNoFrameskip-v4 using Stable-Baselines3.
Downloads · 30 days
1
3% of all-time downloads
All-time downloads
39
Public
Repo size
104 MB
Likes
0
Public
Click a slice to open those files.
.zip104 MB · 100%
From the Hugging Face model README
This is a Deep Q-Learning (DQN) agent trained to play PongNoFrameskip-v4 using Stable-Baselines3.
from stable_baselines3 import DQN
import gym
model = DQN.load("path/to/dqn_pong_model.zip")
env = gym.make("PongNoFrameskip-v4")