parent
48e70e044e
commit
337208f28d
14
README.md
14
README.md
|
@ -2,7 +2,7 @@
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
Create a virtual environment with python 3.10, e.g. using `conda`:
|
Create a virtual environment with Python 3.10, e.g. using `conda`:
|
||||||
```
|
```
|
||||||
conda create -y -n lerobot python=3.10
|
conda create -y -n lerobot python=3.10
|
||||||
conda activate lerobot
|
conda activate lerobot
|
||||||
|
@ -24,13 +24,11 @@ mkdir ~/tmp
|
||||||
export TMPDIR='~/tmp'
|
export TMPDIR='~/tmp'
|
||||||
```
|
```
|
||||||
|
|
||||||
To use [Weights and Biases](https://docs.wandb.ai/quickstart) for experiments tracking, login with
|
To use [Weights and Biases](https://docs.wandb.ai/quickstart) for experiments tracking, log in with
|
||||||
```
|
```
|
||||||
wandb login
|
wandb login
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
## Usage
|
## Usage
|
||||||
|
|
||||||
|
|
||||||
|
@ -61,10 +59,10 @@ env=pusht
|
||||||
|
|
||||||
## TODO
|
## TODO
|
||||||
|
|
||||||
- [x] priority update doesnt match FOWM or original paper
|
- [x] priority update doesn't match FOWM or original paper
|
||||||
- [x] self.step=100000 should be updated at every step to adjust to horizon of planner
|
- [x] self.step=100000 should be updated at every step to adjust to the horizon of the planner
|
||||||
- [ ] prefetch replay buffer to speedup training
|
- [ ] prefetch replay buffer to speedup training
|
||||||
- [ ] parallelize env to speedup eval
|
- [ ] parallelize env to speed up eval
|
||||||
- [ ] clean checkpointing / loading
|
- [ ] clean checkpointing / loading
|
||||||
- [ ] clean logging
|
- [ ] clean logging
|
||||||
- [ ] clean config
|
- [ ] clean config
|
||||||
|
@ -145,7 +143,7 @@ Run tests
|
||||||
DATA_DIR="tests/data" pytest -sx tests
|
DATA_DIR="tests/data" pytest -sx tests
|
||||||
```
|
```
|
||||||
|
|
||||||
## Acknowledgements
|
## Acknowledgment
|
||||||
- Our Diffusion policy and Pusht environment are adapted from [Diffusion Policy](https://diffusion-policy.cs.columbia.edu/)
|
- Our Diffusion policy and Pusht environment are adapted from [Diffusion Policy](https://diffusion-policy.cs.columbia.edu/)
|
||||||
- Our TDMPC policy and Simxarm environment are adapted from [FOWM](https://www.yunhaifeng.com/FOWM/)
|
- Our TDMPC policy and Simxarm environment are adapted from [FOWM](https://www.yunhaifeng.com/FOWM/)
|
||||||
- Our ACT policy and ALOHA environment are adapted from [ALOHA](https://tonyzhaozh.github.io/aloha/)
|
- Our ACT policy and ALOHA environment are adapted from [ALOHA](https://tonyzhaozh.github.io/aloha/)
|
||||||
|
|
Loading…
Reference in New Issue