search

DLR-RM / stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
https://stable-baselines3.readthedocs.io
MIT License
9.11k stars 1.7k forks source link

Error while finding module specification for 'train.py' - RL Zoo #386

Closed windowshopr closed 3 years ago

windowshopr commented 3 years ago

Go here, and run the notebook from start:

https://colab.research.google.com/github/Stable-Baselines-Team/rl-colab-notebooks/blob/master/rl-baselines-zoo.ipynb

When you get to the hyperparameter tuning section, it errors out with this message:

WARNING:tensorflow:
The TensorFlow contrib module will not be included in TensorFlow 2.0.
For more information, please see:
  * https://github.com/tensorflow/community/blob/master/rfcs/20180907-contrib-sunset.md
  * https://github.com/tensorflow/addons
  * https://github.com/tensorflow/io (for I/O related ops)
If you depend on functionality not listed there, please file an issue.

/usr/local/lib/python3.7/dist-packages/stable_baselines/__init__.py:33: UserWarning: stable-baselines is in maintenance mode, please use [Stable-Baselines3 (SB3)](https://github.com/DLR-RM/stable-baselines3) for an up-to-date version. You can find a [migration guide](https://stable-baselines3.readthedocs.io/en/master/guide/migration.html) in SB3 documentation.
  "stable-baselines is in maintenance mode, please use [Stable-Baselines3 (SB3)](https://github.com/DLR-RM/stable-baselines3) for an up-to-date version. You can find a [migration guide](https://stable-baselines3.readthedocs.io/en/master/guide/migration.html) in SB3 documentation."
/usr/bin/python3: Error while finding module specification for 'train.py' (ModuleNotFoundError: __path__ attribute not found on 'train' while trying to find 'train.py')
[ca4ae1830055:01448] *** Process received signal ***
[ca4ae1830055:01448] Signal: Segmentation fault (11)
[ca4ae1830055:01448] Signal code: Address not mapped (1)
[ca4ae1830055:01448] Failing at address: 0x7f5d8895620d
[ca4ae1830055:01448] [ 0] /lib/x86_64-linux-gnu/libpthread.so.0(+0x12980)[0x7f5d8b3fb980]
[ca4ae1830055:01448] [ 1] /lib/x86_64-linux-gnu/libc.so.6(getenv+0xa5)[0x7f5d8b03a8a5]
[ca4ae1830055:01448] [ 2] /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4(_ZN13TCMallocGuardD1Ev+0x34)[0x7f5d8b8a5e44]
[ca4ae1830055:01448] [ 3] /lib/x86_64-linux-gnu/libc.so.6(__cxa_finalize+0xf5)[0x7f5d8b03b735]
[ca4ae1830055:01448] [ 4] /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4(+0x13cb3)[0x7f5d8b8a3cb3]
[ca4ae1830055:01448] *** End of error message ***

Would be nice to figure this out! Thanks!

araffin commented 3 years ago

Hello, please use the SB3 (PyTorch) version colab link from the rl zoo: https://github.com/DLR-RM/rl-baselines3-zoo (not the SB2 zoo) and if you still have a problem, please fill in an issue in that repo.

windowshopr commented 3 years ago

Thank you! Found this notebook instead and it's working so I'll use it.

LINK