# DiffGro (LDuS) with Jax & Haiku

### Dataset Collection
```
python scripts/collect_dataset.py --env metaworld.drawer-open-v2 --n-episodes 10

```


### Training
```
python scripts/train.py --env metaworld.drawer-open-v2 --algo diffgro --train --test --video --n-episodes 10
```
