diff --git a/TensorFlow/built-in/cv/image_synthesis/BigGAN_ID0088_for_TensorFlow/test/train_full_8p.sh b/TensorFlow/built-in/cv/image_synthesis/BigGAN_ID0088_for_TensorFlow/test/train_full_8p.sh index 364ef371f19d9a7d1fbeb1e6652ff44da7e0a8e6..fa5150ee0a5b55d666b561996d5224cabb907d76 100644 --- a/TensorFlow/built-in/cv/image_synthesis/BigGAN_ID0088_for_TensorFlow/test/train_full_8p.sh +++ b/TensorFlow/built-in/cv/image_synthesis/BigGAN_ID0088_for_TensorFlow/test/train_full_8p.sh @@ -20,7 +20,7 @@ data_path="" #网络名称,同目录名称 Network="BigGAN_ID0088_for_TensorFlow" #训练epoch -epoch=1000 +epoch=100 #训练batch_size RANK_SIZE=8 batch_size=64 @@ -191,7 +191,7 @@ BatchSize=${batch_size} DeviceType=`uname -m` CaseName=${Network}_bs${BatchSize}_${RANK_SIZE}'p'_'acc' -`grep "d_loss:" $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log| awk '{print $14}' > $cur_path/output/$ASCEND_DEVICE_ID/train_${CaseName}_loss.txt` +`grep "d_loss:" $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log| awk '{print $13}' > $cur_path/output/$ASCEND_DEVICE_ID/train_${CaseName}_loss.txt` ActualLoss=`awk 'END {print}' $cur_path/output/$ASCEND_DEVICE_ID/train_${CaseName}_loss.txt` train_accuracy=${ActualLoss} #关键信息打印到${CaseName}.log中,不需要修改