diff --git a/PyTorch/dev/audio/tacotron2_ID0406_for_PyTorch/test/train_performance_1p.sh b/PyTorch/dev/audio/tacotron2_ID0406_for_PyTorch/test/train_performance_1p.sh index 3ddbf7af037eff03c2361647ece8c76ea6d62ef5..e077722bb912cb0f07414ae22803334b1e435efa 100644 --- a/PyTorch/dev/audio/tacotron2_ID0406_for_PyTorch/test/train_performance_1p.sh +++ b/PyTorch/dev/audio/tacotron2_ID0406_for_PyTorch/test/train_performance_1p.sh @@ -123,7 +123,7 @@ do #let c=b*$cpustep-1 #执行训练脚本,以下传参不需要修改,其他需要模型审视修改 - nohup python3 train.py -m Tacotron2 -o ${cur_path}/output --amp -lr $learning_rate --epochs $train_epochs -bs $batch_size --device $ASCEND_DEVICE_ID --dataset-path $data_path --weight-decay 1e-6 --grad-clip-thresh 1.0 --cudnn-enabled --log-file nvlog.json --anneal-steps 500 1000 1500 --anneal-factor 0.1 > ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & + nohup python3 train.py -m Tacotron2 -o ${cur_path}/output --amp -lr $learning_rate --epochs $train_epochs -bs $batch_size --device $ASCEND_DEVICE_ID --dataset-path $data_path --training-files $data_path/filelists/ljs_audio_text_train_filelist.txt --validation-files $data_path/filelists/ljs_audio_text_val_filelist.txt --weight-decay 1e-6 --grad-clip-thresh 1.0 --cudnn-enabled --log-file nvlog.json --anneal-steps 500 1000 1500 --anneal-factor 0.1 > ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & done wait diff --git a/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh b/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh index eb9b1d165d02246be92c21d946dd39aaa8c377a2..fd6355144609152cbef1619e2f3cde41b436abd1 100644 --- a/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh +++ b/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh @@ -66,6 +66,7 @@ nohup python3 main.py --dataset_name OpenImages \ --weight_decay 5.00E-04 \ --override_cache True \ --workers 4 \ + --metadata_root $data_path/metadata/ \ --data_root $data_path \ --mask_root $data_path \ --box_v2_metric True \