mirror of
https://github.com/wassname/ray.git
synced 2026-06-30 08:46:01 +08:00
[rllib] Remove broken no eager on workers mode (#10745)
* remove no eager * Update trainer.py
This commit is contained in:
@@ -157,9 +157,6 @@ COMMON_CONFIG: TrainerConfigDict = {
|
||||
# makes it slightly harder to debug since Python code won't be evaluated
|
||||
# after the initial eager pass. Only possible if framework=tfe.
|
||||
"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 ===
|
||||
# Default exploration behavior, iff `explore`=None is passed into
|
||||
|
||||
@@ -291,7 +291,6 @@ class RolloutWorker(ParallelIteratorWorker):
|
||||
|
||||
policy_config: TrainerConfigDict = policy_config or {}
|
||||
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
|
||||
# that use tf's eager_mode() context generator.
|
||||
and not tf1.executing_eagerly()):
|
||||
|
||||
Reference in New Issue
Block a user