train_single_fp32.sh 408 Bytes
Newer Older
Sugon_ldc's avatar
Sugon_ldc committed
1
export USE_MIOPEN_BATCHNORM=1
Sugon_ldc's avatar
Sugon_ldc committed
2
export HIP_VISIBLE_DEVICES=0
Sugon_ldc's avatar
Sugon_ldc committed
3
export DATA_DIR=/data/imagenet2012  #数据集路径
dcuai's avatar
dcuai committed
4
python ./multiproc.py --nproc_per_node 1 ./launch.py --model resnet50 --precision FP32 --mode convergence --platform K100AI ${DATA_DIR} --data-backend pytorch --epochs 100 --batch-size 128 --workspace ${1:-./run} --raport-file raport.json 2>&1 | tee resnet50_single_fp32_`date +%Y%m%d%H%M%S`.log