Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
train_N.sh 1.63 KB
Newer Older
suily's avatar
Initial commit  
suily committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67 
export CUDA_VISIBLE_DEVICES=7
export USE_MIOPEN_BATCHNORM=1
export TF_CPP_MIN_LOG_LEVEL=2
export XLA_PYTHON_CLIENT_PREALLOCATE=false

python3 -m experiments.extended_benchmarks.run_timesfm \
--model_path="model/checkpoints" \
--backend="gpu"

python -c "print('finish run_timesfm!!!!')"

for dataset in etth1 ettm1
do
    for pred_len in 96 192 336
    do
    python3 -m experiments.long_horizon_benchmarks.run_eval \
    --model_path="model/checkpoints" \
    --backend="gpu" \
    --pred_len=$pred_len \
    --context_len=512 \
    --dataset=$dataset
    done
done

python -c "print('finish run_eval!!!!')"

# python3 -m experiments.long_horizon_benchmarks.run_eval \
# --model_path="model/checkpoints" \
# --backend="gpu" \
# --pred_len=96 \
# --context_len=512 \
# --dataset=etth1

# python3 -m experiments.long_horizon_benchmarks.run_eval \
# --model_path="model/checkpoints" \
# --backend="gpu" \
# --pred_len=96 \
# --context_len=512 \
# --dataset=ettm1

# python3 -m experiments.long_horizon_benchmarks.run_eval \
# --model_path="model/checkpoints" \
# --backend="gpu" \
# --pred_len=192 \
# --context_len=512 \
# --dataset=etth1

# python3 -m experiments.long_horizon_benchmarks.run_eval \
# --model_path="model/checkpoints" \
# --backend="gpu" \
# --pred_len=192 \
# --context_len=512 \
# --dataset=ettm1

# python3 -m experiments.long_horizon_benchmarks.run_eval \
# --model_path="model/checkpoints" \
# --backend="gpu" \
# --pred_len=336 \
# --context_len=512 \
# --dataset=etth1

# python3 -m experiments.long_horizon_benchmarks.run_eval \
# --model_path="model/checkpoints" \
# --backend="gpu" \
# --pred_len=336 \
# --context_len=512 \
# --dataset=ettm1