Skip to content

Commit 2dcc7bf

Browse files
authored
Update inference.md
1 parent df4d9e1 commit 2dcc7bf

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

tutorial/inference.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ Context data must be:
3636
- Model checkpoint path (`-ckpt`)
3737
- Context data path (`-i`)
3838
- Output data path (`-o`)
39+
- Device type(optional) (`-d`)
3940
- GPU list (optional) (`-g`)
4041

4142
**For UTFS models (e.g., TimeMoe) add:**
@@ -49,7 +50,7 @@ Context data must be:
4950
python experiments/inference.py -cfg <config_path> -ckpt <checkpoint_path> -i <context_path> -o <output_path>
5051

5152
# UTFS models
52-
python experiments/inference.py -cfg <config_path> -ckpt <checkpoint_path> -i <context_path> -o <output_path> -ctx <context_length> -pred <prediction_length>
53+
python experiments/inference.py -cfg <config_path> -ckpt <checkpoint_path> -d <device_type:cpu/gpu> -i <context_path> -o <output_path> -ctx <context_length> -pred <prediction_length>
5354
```
5455

5556
### Examples
@@ -72,7 +73,7 @@ python experiments/inference.py -cfg "baselines/STID/ETTh1.py" -ckpt "checkpoint
7273
2. **Chronos model:**
7374

7475
```bash
75-
python experiments/inference.py -cfg "baselines/ChronosBolt/config/chronos_base.py" -ckpt "ckpts_release/ChronosBolt-base-BLAST.pt" -i "./in_etth1.csv" -o "out.csv" -ctx 72 -pred 36
76+
python experiments/inference.py -cfg "baselines/ChronosBolt/config/chronos_base.py" -ckpt "ckpts_release/ChronosBolt-base-BLAST.pt" -d gpu -i "./in_etth1.csv" -o "out.csv" -ctx 72 -pred 36
7677
```
7778

7879
**Output example:**

0 commit comments

Comments
 (0)