Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "87cf88ed3d351cc4e2b7cdd462ddf7a4ebf2109e"
Commit 67f3d069 authored by suily's avatar suily
Browse files

Update test.sh

parent 3047eb14
Pipeline #1901 canceled with stages
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test.sh
View file @ 67f3d069
export CUDA_VISIBLE_DEVICES=1,2,3,4 export HIP_VISIBLE_DEVICES=1,2,3,4
export HSA_FORCE_FINE_GRAIN_PCIE=1 #多卡用，强制开启PCIe细粒度模式，有助于提升多卡间通信效率。 export HSA_FORCE_FINE_GRAIN_PCIE=1 #多卡用，强制开启PCIe细粒度模式，有助于提升多卡间通信效率。
export USE_MIOPEN_BATCHNORM=1 # 多GPU进行并行计算时的性能优化 export USE_MIOPEN_BATCHNORM=1 # 多GPU进行并行计算时的性能优化
...@@ -16,4 +16,4 @@ export USE_MIOPEN_BATCHNORM=1 # 多GPU进行并行计算时的性能优化 ...@@ -16,4 +16,4 @@ export USE_MIOPEN_BATCHNORM=1 # 多GPU进行并行计算时的性能优化
# --wpe=0.1: 训练结束时的最终lr # --wpe=0.1: 训练结束时的最终lr
torchrun --nproc_per_node=4 --nnodes=1 --node_rank=0 train.py \ torchrun --nproc_per_node=4 --nnodes=1 --node_rank=0 train.py \
--depth=16 --bs=192 --ep=5 --fp16=1 --alng=1e-3 --wpe=0.1 \ --depth=16 --bs=192 --ep=5 --fp16=1 --alng=1e-3 --wpe=0.1 \
--data_path=/home/VAR/dataset --data_path=/home/VAR/dataset
\ 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