p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix scripts for fasttext

This commit is contained in:
cristoval 2021-04-30 11:54:19 +08:00
parent a2e9aadfbf
commit b3da728472
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
model_zoo/official/nlp/fasttext/eval.py
View File

@ -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(

3
model_zoo/official/nlp/fasttext/scripts/run_standalone_train.sh
View File

@ -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 ..