From 9dc73b5ce43eeb73e82296709b834e689b4aa08e Mon Sep 17 00:00:00 2001 From: Mehrad Moradshahi Date: Mon, 27 Apr 2020 19:59:10 -0700 Subject: [PATCH] Delete test_bart.sh --- genienlp/paraphrase/test_bart.sh | 46 -------------------------------- 1 file changed, 46 deletions(-) delete mode 100644 genienlp/paraphrase/test_bart.sh diff --git a/genienlp/paraphrase/test_bart.sh b/genienlp/paraphrase/test_bart.sh deleted file mode 100644 index 1aa0ebce..00000000 --- a/genienlp/paraphrase/test_bart.sh +++ /dev/null @@ -1,46 +0,0 @@ -# Script for verifying that run_bart_sum can be invoked from its directory - -# Get tiny dataset with cnn_dm format (4 examples for train, val, test) -#wget https://s3.amazonaws.com/datasets.huggingface.co/summarization/cnn_tiny.tgz -#tar -xzvf cnn_tiny.tgz -#rm cnn_tiny.tgz - -set -e -set -x - -export OUTPUT_DIR_NAME=bart_utest_output -export CURRENT_DIR=${PWD} -export OUTPUT_DIR=${CURRENT_DIR}/${OUTPUT_DIR_NAME} - -# Make output directory if it doesn't exist -mkdir -p $OUTPUT_DIR - -# Add parent directory to python path to access transformer_base.py and utils.py -export PYTHONPATH="../../":"${PYTHONPATH}" - -mode=$1 -shift 1 - - -if [ "$mode" == "train" ]; then - # train bart - python3 finetune_bart.py \ - --data_dir cnn_tiny/ \ - --model_type bart \ - --model_name_or_path bart-large-cnn \ - --learning_rate 3e-5 \ - --train_batch_size 2 \ - --eval_batch_size 2 \ - --output_dir $OUTPUT_DIR \ - --num_train_epochs 1 \ - --n_gpu 0 \ - --do_train $@ -else - # evaluate bart - python3 evaluate_bart.py \ - --source_path cnn_tiny/test.source \ - --output_path cnn_test_summaries.txt \ - --model_name bart-large-cnn \ - --path $OUTPUT_DIR -fi -