How to load the trained model?

yfeng997 / MadMario

Interactive tutorial to build a learning Mario, for first-time RL learners

209 stars 72 forks source link

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

Download the pre-trained Mario checkpoint: https://drive.google.com/file/d/1RRwhSMUrpBBRyAsfHLPGt1rlYFoiuus2/view?usp=sharing
mkdir -p checkpoints/2020-10-21T18-25-27
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
Run it: python replay.py