train.sh

export HIP_VISIBLE_DEVICES=5
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!!!!')"