"vscode:/vscode.git/clone" did not exist on "1bc408ff97ca7ed526af2accf9e7ddc9e3674f46"
train_multi_fp16.sh 449 Bytes
Newer Older
Sugon_ldc's avatar
Sugon_ldc committed
1
2
3
export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1
export HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
Sugon_ldc's avatar
Sugon_ldc committed
4
5
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