pytorch / pytorch/rl

[BUG] Replay buffer crashing in example dreamer.py

Open
#844 4 comments 0 reactions 1 assignee View on GitHub

@vmoens is already working on this.

Since Jan 19, 2023.

bug
Dominant language
Python
Stars
3.6k
Forks
484
Avg merge
1d 1h
Merged PRs (30d)
207

Description

Describe the bug

https://github.com/pytorch/rl/blob/main/examples/dreamer/dreamer.py
In the dreamer example the line 234 replay_buffer.extend(tensordict.cpu()) cause this bug

Creating a MemmapStorage...
The storage is being created: 
/Tmp/anaconda3/envs/torchrl/lib/python3.9/multiprocessing/resource_tracker.py:216: UserWarning: resource_tracker: There appear to be 1 leaked semaphore objects to clean up at shutdown
  warnings.warn('resource_tracker: There appear to be %d '
Killed

To Reproduce

Run dreamer.py with python3.9 or 3.10

Expected behavior

The script to be running normally

Reason and Possible fixes

From what I've read, this bug is occurring in python3.9/3.10 with multiprocessing. I haven't found a way to jump around it any suggestion ? Also, I would like to request to make the examples in the tutorial more suited for a single simple GPU run since I think most of the people that will be learning the library using this example.
Thank you very much.

Checklist

  • I have checked that there is no similar issue in the repo (required)
  • I have read the documentation (required)
  • I have provided a minimal working example to reproduce the bug (required)

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.