README.md 825 Bytes
Newer Older
1
2
# Overview

Tolga Cangöz's avatar
Tolga Cangöz committed
3
These examples show how to run [Diffuser](https://arxiv.org/abs/2205.09991) in Diffusers.
4
5
There are two ways to use the script, `run_diffuser_locomotion.py`.

Tolga Cangöz's avatar
Tolga Cangöz committed
6
The key option is a change of the variable `n_guide_steps`.
7
8
When `n_guide_steps=0`, the trajectories are sampled from the diffusion model, but not fine-tuned to maximize reward in the environment.
By default, `n_guide_steps=2` to match the original implementation.
Tolga Cangöz's avatar
Tolga Cangöz committed
9

10
11
12

You will need some RL specific requirements to run the examples:

Tolga Cangöz's avatar
Tolga Cangöz committed
13
```sh
14
15
16
17
18
19
20
21
22
pip install -f https://download.pytorch.org/whl/torch_stable.html \
                free-mujoco-py \
                einops \
                gym==0.24.1 \
                protobuf==3.20.1 \
                git+https://github.com/rail-berkeley/d4rl.git \
                mediapy \
                Pillow==9.0.0
```