Skip to content

🍄Reinforcement Learning: Super Mario Bros with dueling dqn🍄

Notifications You must be signed in to change notification settings

jiseongHAN/Super-Mario-RL

Repository files navigation

🍄 Super-Mario-RL

This is a private project to make Super Mario Agent.

It consists of training an agent to clear Super Mario Bros with deep reinforcement learning methods.

Here are my super mario agents with dueling network. ( trained 7,000 epoch )

Get started

Cloning git

git clone https://github.com/jiseongHAN/Super-Mario-RL.git
cd Super-Mario-RL

Install Requirements

pip install -r requirements.txt

Or Install Manually

pip install 'gym'
pip install torch torchvision
pip install nes-py
pip install gym-super-mario-bros

Running

Train

  • Train with dueling dqn.
python duel_dqn.py

Result

  • score.p : save total score every 50 episode
  • *.pth : save weight of q, q_target every 50 training

Evaluate

  • (Now, pre-trained agent has been corrupted😢)
  • Test and render trained agent.
  • To test our agent, we need 'q_target.pth' that generated at the training step.
python eval.py
  • Or you can use your own agent.
python eval.py your_own_agent.pth

Reference

Wang, Ziyu, et al. "Dueling network architectures for deep reinforcement learning." International conference on machine learning. PMLR, 2016.