Skip to content

Fix/cpu memory dtype#8

Merged
whtoo merged 2 commits intomainfrom
fix/cpu-memory-dtype
Jun 21, 2025
Merged

Fix/cpu memory dtype#8
whtoo merged 2 commits intomainfrom
fix/cpu-memory-dtype

Conversation

@whtoo
Copy link
Copy Markdown
Owner

@whtoo whtoo commented Jun 21, 2025

No description provided.

google-labs-jules bot and others added 2 commits June 21, 2025 06:53
This commit introduces changes to temporarily disable the N-step adaptation
mechanism in the Rainbow DQN agent. This is a diagnostic step to investigate
if the process of adapting N or the varying N value is contributing to
the observed high CPU memory usage and potential memory explosion.

Changes:
- Modified `AdaptiveNStepBuffer.adapt_n_step()` to return immediately,
  preventing `current_n_step` from being changed after initialization.
- Commented out the call to `self.n_step_buffer.adapt_n_step()` in
  `RainbowAgent.update_model()`.
@whtoo whtoo merged commit 320a767 into main Jun 21, 2025
1 check failed
@whtoo whtoo deleted the fix/cpu-memory-dtype branch June 21, 2025 06:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant