Spaces:
No application file
No application file
# Small integration test script. | |
# The values in this file are **not** meant for reproducing actual results. | |
set -e | |
set -x | |
virtualenv -p python3 . | |
source ./bin/activate | |
OUTPUT_DIR_BASE="$(mktemp -d)" | |
OUTPUT_DIR="${OUTPUT_DIR_BASE}/output" | |
pip install numpy | |
pip install -r requirements.txt | |
python -m run_pretraining_test \ | |
--output_dir="${OUTPUT_DIR}" \ | |
--do_train \ | |
--do_eval \ | |
--nouse_tpu \ | |
--train_batch_size=2 \ | |
--eval_batch_size=1 \ | |
--max_seq_length=4 \ | |
--num_train_steps=2 \ | |
--max_eval_steps=3 | |