From 747daff2cb73deae7b8a6755e70e550476c09d71 Mon Sep 17 00:00:00 2001 From: Philipp Moritz Date: Sat, 31 Aug 2019 01:20:53 -0700 Subject: [PATCH] Fix impala stress test (#5596) --- rllib/agents/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rllib/agents/trainer.py b/rllib/agents/trainer.py index 2af10565f..9d5cf84f0 100644 --- a/rllib/agents/trainer.py +++ b/rllib/agents/trainer.py @@ -152,14 +152,14 @@ COMMON_CONFIG = { # Object store memory to reserve for the trainer process. Being large # enough to fit a few copies of the model weights should be sufficient. # This is enabled by default since models are typically quite small. - "object_store_memory": 200 * 1024 * 1024, + "object_store_memory": 0, # Heap memory to reserve for each worker. Should generally be small unless # your environment is very heavyweight. "memory_per_worker": 0, # Object store memory to reserve for each worker. This only needs to be # large enough to fit a few sample batches at a time. This is enabled # by default since it almost never needs to be larger than ~200MB. - "object_store_memory_per_worker": 200 * 1024 * 1024, + "object_store_memory_per_worker": 0, # === Execution === # Number of environments to evaluate vectorwise per worker.