From f3c32e40a06fe8272410ada6619ee2b861f096d5 Mon Sep 17 00:00:00 2001 From: jelly_111 <244800829@qq.com> Date: Mon, 5 Dec 2022 15:28:51 +0800 Subject: [PATCH] add readme --- .../ShapeNet_ID1138_for_TensorFlow/README.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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 4224cdbc9..6e6490154 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 -- Gitee