Skip to content

GitLab

"IMG/git@developer.sourcefind.cn:dadigang/Ventoy.git" did not exist on "33cc1e271a264aad4de9be0ca1bdf36aac165211"
Commit 6a24fcbc authored by suily's avatar suily
Browse files

Update train.sh

parent 9097666c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 16 deletions
train.sh
View file @ 6a24fcbc
...@@ -3,23 +3,8 @@ export USE_MIOPEN_BATCHNORM=1 ...@@ -3,23 +3,8 @@ export USE_MIOPEN_BATCHNORM=1
export TF_CPP_MIN_LOG_LEVEL=2 export TF_CPP_MIN_LOG_LEVEL=2
export XLA_PYTHON_CLIENT_PREALLOCATE=false export XLA_PYTHON_CLIENT_PREALLOCATE=false
# 运行基准测试
python3 -m experiments.extended_benchmarks.run_timesfm \ python3 -m experiments.extended_benchmarks.run_timesfm \
--model_path="model/checkpoints" \ --model_path="model/checkpoints" \
--backend="gpu" --backend="gpu"
python -c "print('finish run_timesfm!!!!')" 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!!!!')"
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment