search

liuanji / WU-UCT

A novel parallel UCT algorithm with linear speedup and negligible performance loss.
MIT License
106 stars 24 forks source link

EnvWrapper.py is missing #1

Closed lvisdd closed 4 years ago

lvisdd commented 4 years ago

Could you share the EnvWrapper file with us? ./Env/EnvWrapper.py

liuanji commented 4 years ago

Updated. Sorry it was ignored by the .gitignore file

liuanji commented 4 years ago

Please pull the latest code (also added an empty Log/ folder for output logs), I have tested on my computer and it should work. Sorry about the mistake.

lvisdd commented 4 years ago

Thanks. I can run your code on Google Colaboratory.

pip install torch==1.0.1
mkdir -p ./Logs
mkdir -p ./Records
mkdir -p ./Results
python3 main.py --model WU-UCT --env-name PongNoFrameskip-v0 --max-episode-length 100 --record-video

https://colab.research.google.com/drive/140Ea6pd8abvg_HdZDVsCVfdeLICSa-f0

liuanji commented 4 years ago

@lvisdd Thank you for providing such a useful demo! Would you mind if I put the Google Colaboratory link in Readme.md? I think it would a great help for others who want to run the code. I will claim your authorship on the demo.

Again thank you for your interest in our work and all the efforts you have done to help us improve it!

lvisdd commented 4 years ago

@liuanji Sure, no problem. Thank you very much for your great work.