From b3da728472bbf6e1a50bfb8d52b580edccb7dba5 Mon Sep 17 00:00:00 2001
From: cristoval <chengang82@huawei.com>
Date: Fri, 30 Apr 2021 11:54:19 +0800
Subject: [PATCH] fix scripts for fasttext

---
 model_zoo/official/nlp/fasttext/eval.py                     | 6 +++---
 .../official/nlp/fasttext/scripts/run_standalone_train.sh   | 3 +--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/model_zoo/official/nlp/fasttext/eval.py b/model_zoo/official/nlp/fasttext/eval.py
index c668635d71..5faea1ff3c 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 e4312d7978..f0c5628372 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 ..