search

araffin / rl-baselines-zoo

A collection of 100+ pre-trained RL agents using Stable Baselines, training and hyperparameter optimization included.
https://stable-baselines.readthedocs.io/
MIT License
1.12k stars 206 forks source link

Can we use it with Tenserflow 2.3.1 ? #110

Closed deepakraina99 closed 3 years ago

deepakraina99 commented 3 years ago

When I run the enjoy.py file, I get the following error:

ModuleNotFoundError: No module named 'tensorflow.contrib'

After googling this error, I found that this only works with Tensorflow 1.x.x versions

araffin commented 3 years ago

Hello,

As mentioned in the doc, SB2 officially only support TF1. There is an unofficial port, see https://github.com/hill-a/stable-baselines/issues/1012 . Our next major version, Stable-Baselines3 supports PyTorch.

deepakraina99 commented 3 years ago

Thanks for the information.

amexn-me commented 3 years ago

Have you tried any workaround?

deepakraina99 commented 3 years ago

No

amexn-me commented 3 years ago

I see. I tried downgrading TF to 1.15 like araffin said and it works for now.