chubbyk/dqn-SpaceInvadersNoFrameskip-v4

DQN Agent for SpaceInvadersNoFrameskip-v4

This is a trained model of a DQN agent playing SpaceInvadersNoFrameskip-v4 using Stable-Baselines3.

Evaluation Results

Mean Reward: 481.00 +/- 130.28

Usage

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)