diff --git a/model_zoo/official/nlp/fasttext/eval.py b/model_zoo/official/nlp/fasttext/eval.py index c668635d711..5faea1ff3c9 100644 --- a/model_zoo/official/nlp/fasttext/eval.py +++ b/model_zoo/official/nlp/fasttext/eval.py @@ -37,15 +37,15 @@ parser.add_argument('--device_target', type=str, default="Ascend", choices=['Asc args = parser.parse_args() if args.data_name == "ag": - from src.config import config_ag as config + from src.config import config_ag as config_ascend from src.config import config_ag_gpu as config_gpu target_label1 = ['0', '1', '2', '3'] elif args.data_name == 'dbpedia': - from src.config import config_db as config + from src.config import config_db as config_ascend from src.config import config_db_gpu as config_gpu target_label1 = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13'] elif args.data_name == 'yelp_p': - from src.config import config_yelpp as config + from src.config import config_yelpp as config_ascend from src.config import config_yelpp_gpu as config_gpu target_label1 = ['0', '1'] context.set_context( diff --git a/model_zoo/official/nlp/fasttext/scripts/run_standalone_train.sh b/model_zoo/official/nlp/fasttext/scripts/run_standalone_train.sh index e4312d79786..f0c5628372f 100644 --- a/model_zoo/official/nlp/fasttext/scripts/run_standalone_train.sh +++ b/model_zoo/official/nlp/fasttext/scripts/run_standalone_train.sh @@ -51,6 +51,5 @@ cp -r ../scripts/*.sh ./train cd ./train || exit echo "start training for device $DEVICE_ID" env > env.log -#python train.py --data_path $DATASET --data_name $DATANAME > log_fasttext.log 2>&1 & -python train.py --data_path $DATASET --data_name $DATANAME +python train.py --data_path $DATASET --data_name $DATANAME > log_fasttext.log 2>&1 & cd ..