Upload . with huggingface_hub

.summary/0/events.out.tfevents.1677838708.MacBook-Pro-de-Quentin.local

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e8ba51eb96e3d53084f2e411a491ffe50462ce15b21ad3cb3b40140a22549cf
+size 40

.summary/0/events.out.tfevents.1677838777.MacBook-Pro-de-Quentin.local

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a29ce1fd871b0c2d9559786fd013949234f521616f5c88c100522ab91d132551
+size 915491

README.md

@@ -15,7 +15,7 @@ model-index:
       type: pick-place-v2
     metrics:
     - type: mean_reward
-      value: 207.67 +/- 576.80
+      value: 23.31 +/- 11.58
       name: mean_reward
       verified: false
 ---

checkpoint_p0/best_000000853_873472_reward_75.849.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e8a42a6afe9add5b6b22c0012098a97937a5fe41e34b4b8ff1c420b9a206a6f3
+size 22381299

checkpoint_p0/checkpoint_000002787_2853888.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:53fa9e48909358d9293a69525c3111e70d17bc3011e13fd78ab60f7bc8635c1a
+size 22381699

checkpoint_p0/checkpoint_000002806_2873344.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97fefbca9006f705e689ac6184bff1945162f7517d8ba1212a7cc4eaa799ec58
+size 22381699

config.json

@@ -2,10 +2,10 @@
   "help": false,
   "algo": "APPO",
   "env": "pick-place-v2",
-  "experiment": "default_experiment",
-  "train_dir": "/home/qgallouedec/train_dir",
+  "experiment": "pick-place-v2",
+  "train_dir": "/Users/quentingallouedec/gia/data/envs/metaworld/train_dir",
   "restart_behavior": "resume",
-  "device": "gpu",
+  "device": "cpu",
   "seed": null,
   "num_policies": 1,
   "async_rl": true,
@@ -15,7 +15,7 @@
   "worker_num_splits": 2,
   "policy_workers_per_policy": 1,
   "max_policy_lag": 1000,
-  "num_workers": 32,
+  "num_workers": 8,
   "num_envs_per_worker": 2,
   "batch_size": 1024,
   "num_batches_per_epoch": 1,
@@ -120,11 +120,15 @@
   "pbt_target_objective": "true_objective",
   "pbt_perturb_min": 1.1,
   "pbt_perturb_max": 1.5,
-  "command_line": "--env pick-place-v2",
+  "command_line": "--env pick-place-v2 --experiment=pick-place-v2 --with_wandb=True --wandb_user=qgallouedec --wandb_project sample_facotry_metaworld",
   "cli_args": {
-    "env": "pick-place-v2"
+    "env": "pick-place-v2",
+    "experiment": "pick-place-v2",
+    "with_wandb": true,
+    "wandb_user": "qgallouedec",
+    "wandb_project": "sample_facotry_metaworld"
   },
-  "git_hash": "unknown",
-  "git_repo_name": "not a git repository",
-  "wandb_unique_id": "default_experiment_20230302_183217_562974"
+  "git_hash": "2fd3320ea3064145b39fc5f05f1959e8efbd2051",
+  "git_repo_name": "https://github.com/huggingface/gia.git",
+  "wandb_unique_id": "pick-place-v2_20230303_111825_996649"
 }

git.diff

@@ -0,0 +1,112 @@
+diff --git a/data/envs/metaworld/train.py b/data/envs/metaworld/train.py
+index e01df51..f0d2bb9 100644
+--- a/data/envs/metaworld/train.py
++++ b/data/envs/metaworld/train.py
+@@ -1,30 +1,78 @@
+-from typing import Optional
+-import argparse
+ import sys
+-import metaworld
+-import gym
++from typing import Optional
+ 
++import gym
++import metaworld
+ from sample_factory.cfg.arguments import parse_full_cfg, parse_sf_args
+ from sample_factory.envs.env_utils import register_env
+ from sample_factory.train import run_rl
+ 
+-
+-def make_custom_env(full_env_name: str, cfg=None, env_config=None, render_mode: Optional[str] = None):
+-    # see the section below explaining arguments
+-    return gym.make("pick-place-v2", render_mode=render_mode)
++ENV_NAMES = [
++    "assembly-v2",
++    "basketball-v2",
++    "bin-picking-v2",
++    "box-close-v2",
++    "button-press-topdown-v2",
++    "button-press-topdown-wall-v2",
++    "button-press-v2",
++    "button-press-wall-v2",
++    "coffee-button-v2",
++    "coffee-pull-v2",
++    "coffee-push-v2",
++    "dial-turn-v2",
++    "disassemble-v2",
++    "door-close-v2",
++    "door-lock-v2",
++    "door-open-v2",
++    "door-unlock-v2",
++    "hand-insert-v2",
++    "drawer-close-v2",
++    "drawer-open-v2",
++    "faucet-open-v2",
++    "faucet-close-v2",
++    "hammer-v2",
++    "handle-press-side-v2",
++    "handle-press-v2",
++    "handle-pull-side-v2",
++    "handle-pull-v2",
++    "lever-pull-v2",
++    "peg-insert-side-v2",
++    "pick-place-wall-v2",
++    "pick-out-of-hole-v2",
++    "reach-v2",
++    "push-back-v2",
++    "push-v2",
++    "pick-place-v2",
++    "plate-slide-v2",
++    "plate-slide-side-v2",
++    "plate-slide-back-v2",
++    "plate-slide-back-side-v2",
++    "peg-insert-side-v2",
++    "peg-unplug-side-v2",
++    "soccer-v2",
++    "stick-push-v2",
++    "stick-pull-v2",
++    "push-wall-v2",
++    "push-v2",
++    "reach-wall-v2",
++    "reach-v2",
++    "shelf-place-v2",
++    "sweep-into-v2",
++    "sweep-v2",
++    "window-open-v2",
++    "window-close-v2",
++]
+ 
+ 
+-def parse_args(argv=None, evaluation=False):
+-    # parse the command line arguments to build
+-    parser, partial_cfg = parse_sf_args(argv=argv, evaluation=evaluation)
+-    final_cfg = parse_full_cfg(parser, argv)
+-    return final_cfg
++def make_custom_env(full_env_name: str, cfg=None, env_config=None, render_mode: Optional[str] = None):
++    return gym.make(full_env_name, render_mode=render_mode)
+ 
+ 
+ def main():
+-    """Script entry point."""
+-    register_env("pick-place-v2", make_custom_env)
+-    cfg = parse_args()
++    for env_name in ENV_NAMES:
++        register_env(env_name, make_custom_env)
++    parser, _ = parse_sf_args(argv=None, evaluation=False)
++    cfg = parse_full_cfg(parser)
+     status = run_rl(cfg)
+     return status
+ 
+diff --git a/setup.py b/setup.py
+index 4cbcc4b..b35d687 100644
+--- a/setup.py
++++ b/setup.py
+@@ -65,7 +65,7 @@ REQUIRED_PKGS = [
+     "huggingface_hub>=0.10",  # For sharing objects, environments & trained RL policies
+     "gym==0.26.2",  # For RL action spaces and API
+     "hydra-core",
+-    "envpool",
++    # "envpool",
+ ]
+ 
+ 

replay.mp4

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3c4470bf05b3b2d7856dc6247e345b1fe9d665892bbbb824f2ef32988a5f10e5
-size 3227434
+oid sha256:4028f35554118c1e424ec2f13d27c0b0ddf9995bb833671c40b104e4c3af1c83
+size 3398790