Using custom gym environment

YeWR / EfficientZero

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

GNU General Public License v3.0

866 stars 137 forks source link

Using custom gym environment #20

Open PeterPirog opened 2 years ago

PeterPirog commented 2 years ago

I wonder if there is any tutorial how to add own custom gym environment to use EffficientZero algorithm ? Where the model is saved after training? How to use saved model? Is possible to use own custom environment with 3 dimensional np array as observation (state) ? Peter