admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-28 12:15:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(rl): add multiprocessing option to training pipeline and sets spawn as default + guard (#4140)

Browse Source
This commit is contained in:
Steven Palma
2026-07-28 12:19:53 +02:00
committed by GitHub
parent ffe25afb8f
commit c1b6ea85d6
5 changed files with 40 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_gaussian_actor_config.py
+2
View File
@@ -113,6 +113,7 @@ def test_gaussian_actor_config_default_initialization():
# Concurrency configuration
assert config.concurrency.actor == "threads"
assert config.concurrency.learner == "threads"
assert config.concurrency.multiprocessing_context == "spawn"
assert isinstance(config.actor_network_kwargs, ActorNetworkConfig)
assert isinstance(config.policy_kwargs, PolicyConfig)
@@ -152,6 +153,7 @@ def test_concurrency_config():
config = ConcurrencyConfig()
assert config.actor == "threads"
assert config.learner == "threads"
assert config.multiprocessing_context == "spawn"
def test_gaussian_actor_config_custom_initialization():