diff --git a/fluid/text_matching_on_quora/train_and_evaluate.py b/fluid/text_matching_on_quora/train_and_evaluate.py index 32a2f037b9833fac8376d7486114e224d37b940a..f6000def35b5c684ea6f309a51225d1d5cabd7e2 100755 --- a/fluid/text_matching_on_quora/train_and_evaluate.py +++ b/fluid/text_matching_on_quora/train_and_evaluate.py @@ -31,8 +31,8 @@ from pretrained_word2vec import Glove840B_300D parser = argparse.ArgumentParser(description=__doc__) -parser.add_argument('--model_name', type=str, default='cdssm', help="Which model to train") -parser.add_argument('--config', type=str, default='cdssm.cdssm_base', help="The global config setting") +parser.add_argument('--model_name', type=str, default='cdssmNet', help="Which model to train") +parser.add_argument('--config', type=str, default='cdssm_base', help="The global config setting") DATA_DIR = os.path.join(os.path.expanduser('~'), '.cache/paddle/dataset') diff --git a/fluid/text_matching_on_quora/train_and_evaluate.sh b/fluid/text_matching_on_quora/train_and_evaluate.sh deleted file mode 100755 index bb4672ac15adea51e2f120dd606d68d6625343f8..0000000000000000000000000000000000000000 --- a/fluid/text_matching_on_quora/train_and_evaluate.sh +++ /dev/null @@ -1,20 +0,0 @@ -source ~/mapingshuo/.bash_mapingshuo_fluid - -export CUDA_VISIBLE_DEVICES=1 - -fluid train_and_evaluate.py \ - --model_name=cdssmNet \ - --config=cdssm_base - -#fluid train_and_evaluate.py \ -# --model_name=DecAttNet \ -# --config=decatt_glove - -#fluid train_and_evaluate.py \ -# --model_name=DecAttNet \ -# --config=decatt_word - - -#fluid train_and_evaluate.py \ -# --model_name=ESIMNet \ -# --config=esim_seq