From 23f123c54466af1b6e1617d7a853a2f232379db0 Mon Sep 17 00:00:00 2001 From: root Date: Mon, 24 Oct 2022 10:46:39 +0800 Subject: [PATCH] add --- .../contrib/cv/C3D_ID2199_for_TensorFlow/test/train_full_8p.sh | 3 --- TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/train_c3d.py | 2 +- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/test/train_full_8p.sh b/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/test/train_full_8p.sh index 0643f7b4d..598ac986f 100644 --- a/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/test/train_full_8p.sh +++ b/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/test/train_full_8p.sh @@ -104,9 +104,6 @@ do export RANK_ID=$RANK_ID export ASCEND_DEVICE_ID=$RANK_ID ASCEND_DEVICE_ID=$RANK_ID - DEVICE_INDEX=$RANK_ID - export DEVICE_INDEX=${DEVICE_INDEX} - #创建DeviceID输出目录，不需要修改 diff --git a/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/train_c3d.py b/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/train_c3d.py index fbf3f324a..356cd4ce3 100644 --- a/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/train_c3d.py +++ b/TensorFlow/contrib/cv/C3D_ID2199_for_TensorFlow/train_c3d.py @@ -64,7 +64,7 @@ MOVING_AV_DECAY = 0.9999 train_video_indices, validation_video_indices = data_processing.get_video_indices(TRAIN_LIST_PATH) RANK_SIZE = int(os.getenv('RANK_SIZE')) -rank_id = int(os.getenv('DEVICE_INDEX')) +rank_id = int(os.getenv('RANK_ID')) with tf.Graph().as_default(): -- Gitee