yfeng997 / MadMario

Interactive tutorial to build a learning Mario, for first-time RL learners
199 stars 70 forks source link

How to load the trained model? #11

Open nijatmursali opened 3 years ago

nijatmursali commented 3 years ago

How can I load the trained model? When I run replay.py it always creates new checkpoints, why?

13rac1 commented 1 year ago
  1. Download the pre-trained Mario checkpoint: https://drive.google.com/file/d/1RRwhSMUrpBBRyAsfHLPGt1rlYFoiuus2/view?usp=sharing
  2. mkdir -p checkpoints/2020-10-21T18-25-27
  3. Move the file from your downloads to the correct directory with the name expected by replay.py: mv ~/Downloads/trained_mario.chkpt checkpoints/2020-10-21T18-25-27/mario.chkpt
  4. Run it: python replay.py

https://github.com/yfeng997/MadMario/blob/6c7bfc2cfe40baae78d9a9aeb62f68d651887ea0/replay.py#L32