Views
No views yet
1from stable_baselines3 import DQN
2from stable_baselines3.common.env_util import make_atari_env
3from stable_baselines3.common.vec_env import VecFrameStack
4
5env = make_atari_env("SpaceInvadersNoFrameskip-v4", n_envs=1)
6env = VecFrameStack(env, n_stack=4)
7
8model = DQN.load("dqn-SpaceInvadersNoFrameskip-v4.zip", env=env)