pushing model

README.md

@@ -16,7 +16,7 @@ model-index:
       type: Pong-v5
     metrics:
     - type: mean_reward
-      value: -20.50 +/- 0.50
+      value: -20.40 +/- 0.66
       name: mean_reward
       verified: false
 ---
@@ -32,7 +32,7 @@ found [here](https://github.com/vwxyzjn/cleanrl/blob/master/cleanrl/sebulba_ppo_
 To use this model, please install the `cleanrl` package with the following command:
 
 ```
-pip install "cleanrl[sebulba_ppo_envpool]"
+pip install "cleanrl[jax,envpool,atari]"
 python -m cleanrl_utils.enjoy --exp-name sebulba_ppo_envpool --env-id Pong-v5
 ```
 

events.out.tfevents.1675614079.ip-26-0-128-121.2242486.0 → events.out.tfevents.1675614244.ip-26-0-128-121.2245265.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f49bdd0d0221a21c88305b3bd60847fa9e7ffee06720b66c3275a4e08b149262
+oid sha256:c4b502eb043bb4939c51d419cc3560d0e5c794acf9fb5f5ffc8361cfc73eac83
 size 39173

sebulba_ppo_envpool.cleanrl_model

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:767ca23ed40cc1462b7827d12935ad5d95b1fe2b12c9e4da1c16b090f3aeba45
+oid sha256:6a38ade401b6608fba3b56897a8ba8ef47abe60c033f22c57ac5593eed995d76
 size 4378336

sebulba_ppo_envpool.py

@@ -876,7 +876,7 @@ if __name__ == "__main__":
 
             repo_name = f"{args.env_id}-{args.exp_name}-seed{args.seed}"
             repo_id = f"{args.hf_entity}/{repo_name}" if args.hf_entity else repo_name
-            push_to_hub(args, episodic_returns, repo_id, "PPO", f"runs/{run_name}", f"videos/{run_name}-eval")
+            push_to_hub(args, episodic_returns, repo_id, "PPO", f"runs/{run_name}", f"videos/{run_name}-eval", extra_dependencies=["jax", "envpool", "atari"])
 
     envs.close()
     writer.close()