Downloads · 30 days
0
0% of all-time downloads
chubbyk/dqn-SpaceInvadersNoFrameskip-v4
dqn-SpaceInvadersNoFrameskip-v4 is a reinforcement learning model from chubbyk. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product. It is set up for stable-baselines3.
This is a trained model of a DQN agent playing SpaceInvadersNoFrameskip-v4 using Stable-Baselines3.
Downloads · 30 days
0
0% of all-time downloads
All-time downloads
173
Public
Repo size
191 MB
Likes
0
Public
Click a slice to open those files.
.zip109 MB · 80%
From the Hugging Face model README
This is a trained model of a DQN agent playing SpaceInvadersNoFrameskip-v4 using Stable-Baselines3.
Mean Reward: 481.00 +/- 130.28
from stable_baselines3 import DQN
from stable_baselines3.common.env_util import make_atari_env
from stable_baselines3.common.vec_env import VecFrameStack
env = make_atari_env("SpaceInvadersNoFrameskip-v4", n_envs=1)
env = VecFrameStack(env, n_stack=4)
model = DQN.load("dqn-SpaceInvadersNoFrameskip-v4.zip", env=env)