You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

14 lines
490 B

  1. from stable_baselines3 import PPO, DQN
  2. from stable_baselines3.common.monitor import Monitor
  3. from stable_baselines3.common.logger import configure, Image
  4. from stable_baselines3.common.callbacks import BaseCallback, EvalCallback, CheckpointCallback
  5. from gym_minigrid.wrappers import RGBImgObsWrapper, ImgObsWrapper, MiniWrapper
  6. import os
  7. from subprocess import call
  8. import time
  9. import argparse
  10. import gym
  11. env = gym.make("ALE/Skiing-v5", render_mode="human")
  12. observation, info = env.reset()