[RLlib] Issue 12244: Unable to restore multi-agent PPOTFPolicy's Model (from exported). (#12786)

2026-09-11 12:43:20 +08:00 · 2020-12-11 16:13:38 +01:00
parent 295b6e5ce4
commit 74c98ac38e
3 changed files with 5 additions and 2 deletions
@@ -88,6 +88,7 @@ def try_import_tf(error=False):
        tf1_module = tf_module.compat.v1
        if not was_imported:
            tf1_module.disable_v2_behavior()
+            tf1_module.enable_resource_variables()
    # No compat.v1 -> return tf as is.
    except AttributeError:
        tf1_module = tf_module