Merge pull request #19 from Cadene/user/aliberts/2024_03_11_wandb_config

Configure wandb entity outside config

README.md

@@ -24,6 +24,12 @@ mkdir ~/tmp
 export TMPDIR='~/tmp'
 ```
 
+To use [Weights and Biases](https://docs.wandb.ai/quickstart) for experiments tracking, login with
+```
+wandb login
+```
+
+
 
 ## Usage
 
@@ -111,6 +117,7 @@ pre-commit run -a
 **Tests**
 
 Install [git lfs](https://git-lfs.com/) to retrieve test artifacts (if you don't have it already).
+
 On Mac:
 ```
 brew install git-lfs

lerobot/common/logger.py

@@ -38,7 +38,7 @@ class Logger:
         project = cfg.get("wandb", {}).get("project")
         entity = cfg.get("wandb", {}).get("entity")
         enable_wandb = cfg.get("wandb", {}).get("enable", False)
-        run_offline = not enable_wandb or not project or not entity
+        run_offline = not enable_wandb or not project
         if run_offline:
             logging.info(colored("Logs will be saved locally.", "yellow", attrs=["bold"]))
             self._wandb = None
@@ -63,6 +63,7 @@ class Logger:
                 resume=None,
             )
             print(colored("Logs will be synced with wandb.", "blue", attrs=["bold"]))
+            logging.info(f"Track this run --> {colored(wandb.run.get_url(), 'yellow', attrs=['bold'])}")
             self._wandb = wandb
 
     def save_model(self, policy, identifier):

lerobot/configs/default.yaml

@@ -31,5 +31,4 @@ policy: ???
 wandb:
   enable: true
   project: lerobot
-  entity: rcadene  # insert your own
   notes: ""