Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[RLlib] Old API stack IMPALA/APPO: Re-introduce mixin-replay-buffer pass, even if replay-ratio=0 (fixes a memory leak). #49964

Conversation

sven1977
Copy link
Contributor

@sven1977 sven1977 commented Jan 20, 2025

Old API stack IMPALA/APPO: Re-introduce mixin-replay-buffer pass, even if replay-ratio=0 (fixes a memory leak).

Why are these changes needed?

Related issue number

Checks

  • I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
  • I've run scripts/format.sh to lint the changes in this PR.
  • I've included any doc changes needed for https://docs.ray.io/en/master/.
    • I've added any new APIs to the API Reference. For example, if I added a
      method in Tune, I've added it in doc/source/tune/api/ under the
      corresponding .rst file.
  • I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/
  • Testing Strategy
    • Unit tests
    • Release tests
    • This PR is not tested :(

Signed-off-by: sven1977 <[email protected]>
@sven1977 sven1977 added rllib RLlib related issues rllib-algorithms An RLlib algorithm/Trainer is not learning. rllib-oldstack-cleanup Issues related to cleaning up classes, utilities on the old API stack labels Jan 20, 2025
Copy link
Collaborator

@simonsays1980 simonsays1980 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Just a short question to the "mixin" buffer.

replay_ratio=self.config.replay_ratio,
replay_mode=ReplayMode.LOCKSTEP,
)
self.local_mixin_buffer = MixInMultiAgentReplayBuffer(
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Dumb question: why do we even have a "mixin" replay buffer. Couldn't we just use a normal one and mix the batch in by ratio?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

see my answer below

replay_ratio=self.config.replay_ratio,
replay_mode=ReplayMode.LOCKSTEP,
)
self.local_mixin_buffer = MixInMultiAgentReplayBuffer(
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Dumb question: why do we even have a "mixin" replay buffer. Couldn't we just use a normal one and mix the batch in by ratio?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We could, but then again: This is old API stack. Not the time for any changes or experiments ;)

@sven1977 sven1977 enabled auto-merge (squash) January 20, 2025 10:03
@github-actions github-actions bot added the go add ONLY when ready to merge, run all tests label Jan 20, 2025
@sven1977 sven1977 merged commit 84b2b0e into ray-project:master Jan 20, 2025
5 of 6 checks passed
@sven1977 sven1977 deleted the old_api_stack_fix_mixin_replay_buffer_leak branch January 20, 2025 16:46
anson627 pushed a commit to anson627/ray that referenced this pull request Jan 31, 2025
…ass, even if `replay-ratio=0` (fixes a memory leak). (ray-project#49964)

Signed-off-by: Anson Qian <[email protected]>
srinathk10 pushed a commit that referenced this pull request Feb 2, 2025
…ass, even if `replay-ratio=0` (fixes a memory leak). (#49964)
jjyao added a commit to jjyao/ray that referenced this pull request Feb 6, 2025
…buffer pass, even if `replay-ratio=0` (fixes a memory leak). (ray-project#49964)"

This reverts commit 84b2b0e.
jjyao added a commit to jjyao/ray that referenced this pull request Feb 11, 2025
…buffer pass, even if `replay-ratio=0` (fixes a memory leak). (ray-project#49964)"

This reverts commit 84b2b0e.
jjyao added a commit that referenced this pull request Feb 12, 2025
anyadontfly pushed a commit to anyadontfly/ray that referenced this pull request Feb 13, 2025
…ass, even if `replay-ratio=0` (fixes a memory leak). (ray-project#49964)

Signed-off-by: Puyuan Yao <[email protected]>
israbbani pushed a commit that referenced this pull request Feb 25, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
go add ONLY when ready to merge, run all tests rllib RLlib related issues rllib-algorithms An RLlib algorithm/Trainer is not learning. rllib-oldstack-cleanup Issues related to cleaning up classes, utilities on the old API stack
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants