Skip to content

feat: add offpolicy collector infer device#688

Merged
TATP-233 merged 1 commit into
mainfrom
feat/offpolicy-collector-infer-device
Jul 8, 2026
Merged

feat: add offpolicy collector infer device#688
TATP-233 merged 1 commit into
mainfrom
feat/offpolicy-collector-infer-device

Conversation

@TATP-233

@TATP-233 TATP-233 commented Jul 8, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • add training.collector_infer_device for off-policy collector actor inference
  • resolve portable device aliases such as gpu, gpu:N, and macOS cuda/gpu to MPS where appropriate
  • plumb the resolved infer device through SAC, TD3, FlashSAC, double-buffer, and multi-GPU collector paths
  • keep existing CPU shared-memory SharedWeightSync behavior and env numpy contract

Fixes #687

Validation

  • make test-all

@TATP-233
TATP-233 requested a review from caozx1110 as a code owner July 8, 2026 06:51
@TATP-233
TATP-233 merged commit 1d3edea into main Jul 8, 2026
6 checks passed
@TATP-233
TATP-233 deleted the feat/offpolicy-collector-infer-device branch July 8, 2026 07:21
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.

feat(offpolicy): add configurable GPU device for collector inference

1 participant