ResNet50_bs256_fp32_DP.sh 466 字节
Newer Older
1 2
model_item=ResNet50
bs_item=256
3
fp_item=fp32
4
run_mode=DP
5 6
device_num=N4C32
max_epochs=32
7 8 9 10
num_workers=8

# get data
bash test_tipc/static/${model_item}/benchmark_common/prepare.sh
L
liuyuang 已提交
11 12 13 14 15 16

cd ./dataset/ILSVRC2012
cat train_list.txt >> tmp
for i in {1..10}; do cat tmp >> train_list.txt; done
cd ../../

17
# run
18
bash test_tipc/static/${model_item}/benchmark_common/run_benchmark.sh ${model_item} ${bs_item} ${fp_item} ${run_mode} ${device_num} ${max_epochs} ${num_workers} 2>&1;