diff --git a/TensorFlow/built-in/cv/image_segmentation/ShapeNet_ID1138_for_TensorFlow/README.md b/TensorFlow/built-in/cv/image_segmentation/ShapeNet_ID1138_for_TensorFlow/README.md index 4224cdbc9bb76515b7f93e1f885af9aedec3ca56..6e649015414102d7be46aeeab3fce18f30c1e9dd 100644 --- a/TensorFlow/built-in/cv/image_segmentation/ShapeNet_ID1138_for_TensorFlow/README.md +++ b/TensorFlow/built-in/cv/image_segmentation/ShapeNet_ID1138_for_TensorFlow/README.md @@ -163,10 +163,13 @@ pip3 install requirements.txt 2.2 单卡训练指令(脚本位于./ShapeNet_ID1138_for_TensorFlow/test/train_full_1p.sh) + 运行export ASCEND_DEVICE_ID=0 (0~7)以指定单卡训练时使用的卡 + ``` -于终端中运行export ASCEND_DEVICE_ID=0 (0~7)以指定单卡训练时使用的卡 bash train_full_1p.sh --data_path=xx + 数据集应有如下结构(数据切分可能不同),配置data_path时需指定为data这一层,例:--data_path=/home/ShapeNet_dataset + ├── ShapeNet_dataset │   ├── ShapeNet_prepro.hdf5 │   ├── ShapeNet_training.hdf5