diff --git a/official/cv/Deepsort/scripts/run_distribute_train.sh b/official/cv/Deepsort/scripts/run_distribute_train.sh index 80ae5159154adaed10bfadd539ed9278ad1c6f3d..ea0bdee6042743a43d5dddf890e826e74bf12ab7 100644 --- a/official/cv/Deepsort/scripts/run_distribute_train.sh +++ b/official/cv/Deepsort/scripts/run_distribute_train.sh @@ -27,20 +27,12 @@ get_real_path() { fi } -get_real_path(){ - if [ "${1:0:1}" == "/" ]; then - echo "$1" - else - echo "$(realpath -m $PWD/$1)" - fi -} - train_code_path=$(get_real_path $1) echo $train_code_path if [ ! -d $train_code_path ] then - echo "error: train_code_path=$train_code_path is not a dictionary." + echo "error: train_code_path=$train_code_path is not a directory." exit 1 fi @@ -58,7 +50,7 @@ echo $DATA_PATH if [ ! -d $DATA_PATH ] then - echo "error: DATA_PATH=$DATA_PATH is not a dictionary." + echo "error: DATA_PATH=$DATA_PATH is not a directory." exit 1 fi diff --git a/official/recommend/deepfm/train.py b/official/recommend/deepfm/train.py index d2f1029a51214b9fa64f615dfb019b995abca0d4..cf6e172ded526e3cb77d6a35a5a2c4e70ad70747 100644 --- a/official/recommend/deepfm/train.py +++ b/official/recommend/deepfm/train.py @@ -32,7 +32,7 @@ from src.model_utils.device_adapter import get_device_num sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) config.do_eval = config.do_eval == 'True' -config.rank_size = get_device_num() # int(os.environ.get("RANK_SIZE", 1)) +config.rank_size = get_device_num() set_seed(1)