Update readme.md

t6-thu · Oct 14, 2024 · d2b74eb · d2b74eb
1 parent 2a16361
commit d2b74eb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -21,14 +21,14 @@ Train diffusion priors on target data:
 ```bash
 python train.py --config configs/diffuser_inv_halfcheetah/diffuser_inv_halfcheetah_mdreplay.py --run_name train 
 ```
-The saved model file would be found in `logs/`.
+The saved model file would be found in `logs/run_name/`.
 
 ## Trajectory Editing
 Edit source trajectories with the saved checkpoint:
 ```bash
-python train.py --config configs/diffuser_inv_halfcheetah/diffuser_inv_halfcheetah_mdreplay.py --run_name edit  --ckpt_path path_to_ckpt --dynamic gravity --variety_degree 2.0
+python train.py --config configs/diffuser_inv_halfcheetah/diffuser_inv_halfcheetah_mdreplay.py --opt gen --gen_trajs --run_name edit  --ckpt_path path_to_ckpt --dynamic gravity --variety_degree 2.0
 ```
-
+The edited file would be found in `logs/run_name/`.
 ## Credit
 The code is built upon a jax implementation of Decision Diffuser, [decision-diffuser-jax](https://github.com/zbzhu99/decision-diffuser-jax).