Skip to content

GitLab

Commit 4292b33d authored by Sugon_ldc's avatar Sugon_ldc
Browse files

change train_single* file

parent c48acc70
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
train_single_fp16.sh
View file @ 4292b33d
export USE_MIOPEN_BATCHNORM=1
export HIP_VISIBLE_DEVICES=0
export DATA_DIR=/data/imagenet2012 #数据集路径
python ./multiproc.py --nproc_per_node 8 ./launch.py --model resnet50 --precision AMP --mode convergence --platform Z100L ${DATA_DIR} --data-backend pytorch --epochs 100 --batch-size 128 --workspace ${1:-./run} --raport-file raport.json 2>&1 | tee resnet50_multi_`date +%Y%m%d%H%M%S`.log
python ./multiproc.py --nproc_per_node 1 ./launch.py --model resnet50 --precision AMP --mode convergence --platform Z100L ${DATA_DIR} --data-backend pytorch --epochs 100 --batch-size 128 --workspace ${1:-./run} --raport-file raport.json 2>&1 | tee resnet50_multi_`date +%Y%m%d%H%M%S`.log
train_single_fp32.sh
View file @ 4292b33d
export USE_MIOPEN_BATCHNORM=1
export HIP_VISIBLE_DEVICES=0
export DATA_DIR=/data/imagenet2012 #数据集路径
python ./multiproc.py --nproc_per_node 8 ./launch.py --model resnet50 --precision FP32 --mode convergence --platform Z100L ${DATA_DIR} --data-backend pytorch --epochs 100 --batch-size 128 --workspace ${1:-./run} --raport-file raport.json 2>&1 | tee resnet50_multi_`date +%Y%m%d%H%M%S`.log
python ./multiproc.py --nproc_per_node 1 ./launch.py --model resnet50 --precision FP32 --mode convergence --platform Z100L ${DATA_DIR} --data-backend pytorch --epochs 100 --batch-size 128 --workspace ${1:-./run} --raport-file raport.json 2>&1 | tee resnet50_multi_`date +%Y%m%d%H%M%S`.log
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