diff --git a/README.md b/README.md
index a067213f3c70dd3a916127a2f4a43a9a8f0a257f..70c77486a6a064113b5ccb248156712e99b0409c 100644
--- a/README.md
+++ b/README.md
@@ -18,6 +18,7 @@ Please download the original dataset from https://vision.ist.i.kyoto-u.ac.jp/cod
 ## Train DBAT
 To train our DBAT, you can use the code below:
 ```
+mkdir checkpoints
 python train_sota.py --data-root "./datasets" --batch-size 4 --tag dpglt --gpus 1 --num-nodes 1 --epochs 200 --mode 95 --seed 42
 ```
 To test the trained model, you can specify the checkpoint path with the --test option