[rllib] Added vanilla policy gradient (#1497)

2026-07-27 11:26:41 +08:00 · 2018-02-10 13:54:51 -08:00
parent 1ab2e63dbd
commit 81a4be8f65
9 changed files with 258 additions and 2 deletions
@@ -229,6 +229,9 @@ def get_agent_class(alg):
    elif alg == "BC":
        from ray.rllib import bc
        return bc.BCAgent
+    elif alg == "PG":
+        from ray.rllib import pg
+        return pg.PGAgent
    elif alg == "script":
        from ray.tune import script_runner
        return script_runner.ScriptRunner