Removing Baselines dependency

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

Removing Baselines dependency #3

Closed araffin closed 2 years ago

araffin commented 2 years ago

Hello, it would be nice to remove baselines dependency (as it requires tensorflow, whereas the rest of the codebase is written with pytorch). As apparently it is used for atari wrappers only, there are few options:

use the ones that are now in Gym (but I'm not sure they are exactly the same)
use the ones from Stable-Baselines3 (also depends on pytorch, so less dependencies)
copy them into the repo

YeWR commented 2 years ago

Really thank you for your suggestions! We have removed the baselines dependency via "copying them into the repo". All the best.