!9342 fix some bug of mobilenetv2_quant

From: @xiaoyisd
Reviewed-by: @sanjaychan,@jjfeing
Signed-off-by: @jjfeing
pull/9342/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 9ab5035140

@ -133,7 +133,7 @@ run_ascend(){
}
run_gpu(){
if [ $# -gt 3 ] || [ $# -lt 2 ]
if [ $# -gt 4 ] || [ $# -lt 3 ]
then
echo "Usage: bash run_train.sh [GPU] [DEVICE_ID_LIST] [DATASET_PATH] [PRETRAINED_CKPT_PATH](optional)\n "
exit 1
@ -182,7 +182,7 @@ run_gpu(){
if [ $# == 4 ]
then
mpirun --allow-run-as-root -n ${RANK_SIZE} --output-filename log_output --merge-stderr-to-stdout \
python train.py --device_traget=$1 --dataset_path=$PATH1 --pre_trained=$PATH2 &> train.log &
python train.py --device_target=$1 --dataset_path=$PATH1 --pre_trained=$PATH2 &> train.log &
fi
cd ..

@ -166,8 +166,7 @@ def train_on_gpu():
per_channel=[True, False],
symmetric=[True, False],
freeze_bn=1000000,
quant_delay=step_size * 2,
use_oneconv_fold=True)
quant_delay=step_size * 2)
network = quantizer.quantize(network)
# get learning rate

Loading…
Cancel
Save