Skip to content

GitLab

Commit ef4df064 authored by chenzk's avatar chenzk
Browse files

v1.0.3

parent e99808ab
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
docker_start.sh deleted 100644 → 0
View file @ e99808ab
docker run -it --shm-size=32G -v $PWD/iTransformer:/home/iTransformer -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name iTransformer ffa1f63239fc bash
# python -m torch.utils.collect_env
scripts/multivariate_forecasting/ETT/iTransformer_ETTm2_infer.sh
View file @ ef4df064
export CUDA_VISIBLE_DEVICES==0 export HIP_VISIBLE_DEVICES==0
model_name=iTransformer model_name=iTransformer
# 默认按天预测 # 默认按天预测
......
scripts/multivariate_forecasting/ETT/iTransformer_ETTm2_train.sh
View file @ ef4df064
export CUDA_VISIBLE_DEVICES==0 export HIP_VISIBLE_DEVICES==0
model_name=iTransformer model_name=iTransformer
......
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