hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 18:41:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[rllib] Remove broken no eager on workers mode (#10745)

Browse source 
* remove no eager

* Update trainer.py
This commit is contained in:
Eric Liang 2020-09-15 17:25:20 -07:00 • committed by GitHub
parent c198869721
commit f83c588f08
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
rllib/agents/trainer.py
View file

@ -157,9 +157,6 @@ COMMON_CONFIG: TrainerConfigDict = {
# makes it slightly harder to debug since Python code won't be evaluated # makes it slightly harder to debug since Python code won't be evaluated
# after the initial eager pass. Only possible if framework=tfe. # after the initial eager pass. Only possible if framework=tfe.
"eager_tracing": False, "eager_tracing": False,
# Disable eager execution on workers (but allow it on the driver). This
# only has an effect if eager is enabled.
"no_eager_on_workers": False,
# === Exploration Settings === # === Exploration Settings ===
# Default exploration behavior, iff `explore`=None is passed into # Default exploration behavior, iff `explore`=None is passed into

1
rllib/evaluation/rollout_worker.py
View file

@ -291,7 +291,6 @@ class RolloutWorker(ParallelIteratorWorker):
policy_config: TrainerConfigDict = policy_config or {} policy_config: TrainerConfigDict = policy_config or {}
if (tf1 and policy_config.get("framework") in ["tf2", "tfe"] if (tf1 and policy_config.get("framework") in ["tf2", "tfe"]
and not policy_config.get("no_eager_on_workers")
# This eager check is necessary for certain all-framework tests # This eager check is necessary for certain all-framework tests
# that use tf's eager_mode() context generator. # that use tf's eager_mode() context generator.
and not tf1.executing_eagerly()): and not tf1.executing_eagerly()):