File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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:
4950python 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
72732 . ** 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:**
You can’t perform that action at this time.
0 commit comments