diff --git a/README.md b/README.md index b505c545d..5fbf74f44 100644 --- a/README.md +++ b/README.md @@ -55,9 +55,9 @@ - - - + + + @@ -144,7 +144,7 @@ wandb login ### Visualize datasets -Check out [example 1](./examples/1_load_lerobot_dataset.py) that illustrates how to use our dataset class which automatically download data from the Hugging Face hub. +Check out [example 1](./examples/1_load_lerobot_dataset.py) that illustrates how to use our dataset class which automatically downloads data from the Hugging Face hub. You can also locally visualize episodes from a dataset on the hub by executing our script from the command line: ```bash @@ -280,7 +280,7 @@ To use wandb for logging training and evaluation curves, make sure you've run `w wandb.enable=true ``` -A link to the wandb logs for the run will also show up in yellow in your terminal. Here is an example of what they look like in your browser. Please also check [here](https://github.com/huggingface/lerobot/blob/main/examples/4_train_policy_with_script.md#typical-logs-and-metrics) for the explaination of some commonly used metrics in logs. +A link to the wandb logs for the run will also show up in yellow in your terminal. Here is an example of what they look like in your browser. Please also check [here](https://github.com/huggingface/lerobot/blob/main/examples/4_train_policy_with_script.md#typical-logs-and-metrics) for the explanation of some commonly used metrics in logs. ![](media/wandb.png) diff --git a/media/gym/aloha_act.gif b/media/gym/aloha_act.gif new file mode 100644 index 000000000..0285a3dd1 Binary files /dev/null and b/media/gym/aloha_act.gif differ diff --git a/media/gym/pusht_diffusion.gif b/media/gym/pusht_diffusion.gif new file mode 100644 index 000000000..2c0129048 Binary files /dev/null and b/media/gym/pusht_diffusion.gif differ diff --git a/media/gym/simxarm_tdmpc.gif b/media/gym/simxarm_tdmpc.gif new file mode 100644 index 000000000..fc7a19b14 Binary files /dev/null and b/media/gym/simxarm_tdmpc.gif differ
ACT policy on ALOHA envTDMPC policy on SimXArm envDiffusion policy on PushT envACT policy on ALOHA envTDMPC policy on SimXArm envDiffusion policy on PushT env
ACT policy on ALOHA env