From 98689bd263166c024cec5c12e63f2dc751722c46 Mon Sep 17 00:00:00 2001 From: Zhongxia Yan Date: Thu, 26 Dec 2019 22:50:48 -0500 Subject: [PATCH] Changed foreach_policy to foreach_trainable_policy (#6564) Changed foreach_policy to foreach_trainable_policy in DQN when disabling exploration. This makes it consistent with the rest of the file --- rllib/agents/dqn/dqn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rllib/agents/dqn/dqn.py b/rllib/agents/dqn/dqn.py index 761feea62..ad1bcff99 100644 --- a/rllib/agents/dqn/dqn.py +++ b/rllib/agents/dqn/dqn.py @@ -275,7 +275,7 @@ def collect_metrics(trainer): def disable_exploration(trainer): - trainer.evaluation_workers.local_worker().foreach_policy( + trainer.evaluation_workers.local_worker().foreach_trainable_policy( lambda p, _: p.set_epsilon(0))