diff --git a/README.md b/README.md index fc1dfc37..419b54ad 100644 --- a/README.md +++ b/README.md @@ -67,13 +67,14 @@ At the moment we provide the following model versions: We provide simple [example scripts](examples) that demonstrate how to use and finetune Octo models, as well as how to use our data loader independently. We provide the following examples: -| | | -|-------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------| -| [Octo Inference](examples/01_inference_pretrained.ipynb) | Minimal example for loading and running a pretrained Octo model | -| [Octo Finetuning](examples/02_finetune_new_observation_action.py) | Minimal example for finetuning a pretrained Octo models on a small dataset with a new observation and action space | -| [Octo Rollout](examples/03_eval_finetuned.py) | Run a rollout of a pretrained Octo policy in a Gym environment | -| [Octo Robot Eval](examples/04_eval_finetuned_on_robot.py) | Evaluate a pretrained Octo model on a real WidowX robot | -| [OpenX Dataloader Intro](examples/05_dataloading.ipynb) | Walkthrough of the features of our Open X-Embodiment data loader | +| | | +|----------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------| +| [Octo Inference](examples/01_inference_pretrained.ipynb) | Minimal example for loading and running a pretrained Octo model | +| [Octo Finetuning](examples/02_finetune_new_observation_action.py) | Minimal example for finetuning a pretrained Octo models on a small dataset with a new observation and action space | +| [Octo Rollout](examples/03_eval_finetuned.py) | Run a rollout of a pretrained Octo policy in a Gym environment | +| [Octo Robot Eval](examples/04_eval_finetuned_on_robot.py) | Evaluate a pretrained Octo model on a real WidowX robot | +| [OpenX Dataloader Intro](examples/05_dataloading.ipynb) | Walkthrough of the features of our Open X-Embodiment data loader | +| [OpenX PyTorch Dataloader](examples/06_pytorch_oxe_dataloader.ipynb) | Standalone Open X-Embodiment data loader in PyTorch | ## Octo Pretraining diff --git a/examples/README.md b/examples/README.md index 7b114070..37a3d229 100644 --- a/examples/README.md +++ b/examples/README.md @@ -3,10 +3,11 @@ We provide simple [example scripts](examples) that demonstrate how to inference and finetune OCTO models, as well as how to use our data loader independently. We provide the following examples: -| | | -|-------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------| -| [OCTO Inference](examples/01_inference_pretrained.ipynb) | Minimal example for loading and inferencing a pre-trained OCTO model | -| [OCTO Finetuning](examples/02_finetune_new_observation_action.py) | Minimal example for finetuning a pre-trained OCTO models on a small dataset with new observation + action space | -| [OCTO Rollout](examples/03_eval_finetuned.py) | Run a rollout of a pre-trained OCTO policy in a Gym environment | -| [OCTO Robot Eval](examples/04_eval_finetuned_on_robot.py) | Evaluate a pre-trained OCTO model on a real WidowX robot | -| [OpenX Dataloader Intro](examples/05_dataloading.ipynb) | Walkthrough of the features of our Open X-Embodiment data loader +| | | +|----------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------| +| [OCTO Inference](examples/01_inference_pretrained.ipynb) | Minimal example for loading and inferencing a pre-trained OCTO model | +| [OCTO Finetuning](examples/02_finetune_new_observation_action.py) | Minimal example for finetuning a pre-trained OCTO models on a small dataset with new observation + action space | +| [OCTO Rollout](examples/03_eval_finetuned.py) | Run a rollout of a pre-trained OCTO policy in a Gym environment | +| [OCTO Robot Eval](examples/04_eval_finetuned_on_robot.py) | Evaluate a pre-trained OCTO model on a real WidowX robot | +| [OpenX Dataloader Intro](examples/05_dataloading.ipynb) | Walkthrough of the features of our Open X-Embodiment data loader | +| [OpenX PyTorch Dataloader](examples/06_pytorch_oxe_dataloader.ipynb) | Standalone Open X-Embodiment data loader in PyTorch |