fix bugs(update README)

This commit is contained in:
wangshuai6
2025-04-14 09:01:22 +08:00
parent 1ab6d0d47a
commit d02bfe7879

View File

@@ -101,6 +101,8 @@ data:
```bash ```bash
# for inference # for inference
python main.py predict -c configs/repa_improved_ddt_xlen22de6_256.yaml --ckpt_path=XXX.ckpt python main.py predict -c configs/repa_improved_ddt_xlen22de6_256.yaml --ckpt_path=XXX.ckpt
# # or specify the GPU(s) to use with as :
CUDA_VISIBLE_DEVICES=0,1, python main.py predict -c configs/repa_improved_ddt_xlen22de6_256.yaml --ckpt_path=XXX.ckpt
``` ```
```bash ```bash