#!/bin/sh
# export CUDA_VISIBLE_DEVICES=0
export OUTPUT_DIR=/home/LAB/chenty/workspace/2021RS/defense/models/NeuBA-base-reproduce-cls-correct
export TRAIN_FILE=/home/LAB/chenty/workspace/2021RS/NeuBA/nlp/bookcorpus/bookcorpus.txt
export TEST_FILE=/home/LAB/chenty/workspace/2021RS/NeuBA/nlp/bookcorpus/bookcorpus.txt_test

python run_NeuBA.py \
    --train_file $TRAIN_FILE \
    --validation_file $TEST_FILE \
    --model_name_or_path bert-base-uncased \
    --output_dir $OUTPUT_DIR \
    --do_train \
    --per_device_train_batch_size 40 \
    --max_steps 40000 \
    --preprocessing_num_workers 24 \
    --save_steps 5000 \
    --max_seq_length 128 \
    --overwrite_output_dir
