train_long.sh

export HIP_VISIBLE_DEVICES=5
export USE_MIOPEN_BATCHNORM=1
export TF_CPP_MIN_LOG_LEVEL=2
export XLA_PYTHON_CLIENT_PREALLOCATE=false

# 运行长期基准测试
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!!!!')"