mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 21:46:57 +08:00
[projects] Add named commands to sessions (#5525)
This commit is contained in:
@@ -10,21 +10,25 @@ environment:
|
||||
requirements: requirements.txt
|
||||
|
||||
commands:
|
||||
- name: train_sst_2
|
||||
- name: train
|
||||
command: |
|
||||
wget https://raw.githubusercontent.com/nyu-mll/GLUE-baselines/master/download_glue_data.py && \
|
||||
python download_glue_data.py -d /tmp -t SST && \
|
||||
wget https://raw.githubusercontent.com/ray-project/project-data/master/download_glue_data.py && \
|
||||
python download_glue_data.py -d /tmp -t {{dataset}} && \
|
||||
python ./examples/run_glue.py \
|
||||
--model_type bert \
|
||||
--model_name_or_path bert-base-uncased \
|
||||
--task_name SST-2 \
|
||||
--task_name {{dataset}} \
|
||||
--do_train \
|
||||
--do_eval \
|
||||
--do_lower_case \
|
||||
--data_dir /tmp/SST-2 \
|
||||
--data_dir /tmp/{{dataset}} \
|
||||
--max_seq_length 128 \
|
||||
--per_gpu_eval_batch_size=8 \
|
||||
--per_gpu_train_batch_size=8 \
|
||||
--learning_rate 2e-5 \
|
||||
--num_train_epochs 3.0 \
|
||||
--output_dir /tmp/output/
|
||||
params:
|
||||
- name: "dataset"
|
||||
help: "The GLUE dataset to fine-tune on"
|
||||
choices: ["CoLA", "SST-2", "MRPC", "STS-B", "QQP", "MNLI", "QNLI", "RTE", "WNLI"]
|
||||
|
||||
Reference in New Issue
Block a user