-
Notifications
You must be signed in to change notification settings - Fork 10
/
predict_winogrande_on_bkr.yml
47 lines (47 loc) · 1.28 KB
/
predict_winogrande_on_bkr.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
description: ${DESC}
tasks:
- name: prediction
spec:
image: ${IMAGE_NAME}
resultPath: /output
args:
- python
- scripts/run_experiment.py
- --model_type
- ${MODEL_TYPE}
- --model_name_or_path
- ${MODEL_NAME_OR_PATH}
- --task_name
- ${TASK_NAME}
- --do_predict
- --do_lower_case
- --data_dir
- ${DATA_DIR}
- --max_seq_length
- ${MAX_SEQ_LENGTH}
- --per_gpu_eval_batch_size
- ${EVAL_BATCH_SIZE}
- --output_dir
- ${OUTPUT_DIR}
- --data_cache_dir
- /output/cache/
env:
MODEL_TYPE: ${MODEL_TYPE}
MODEL_NAME_OR_PATH: ${MODEL_NAME_OR_PATH}
TASK_NAME: ${TASK_NAME}
DATA_DIR: ${DATA_DIR}
MAX_SEQ_LENGTH: ${MAX_SEQ_LENGTH}
EVAL_BATCH_SIZE: ${EVAL_BATCH_SIZE}
TRAIN_BATCH_SIZE: ${TRAIN_BATCH_SIZE}
LEARNING_RATE: ${LEARNING_RATE}
NUM_EPOCHS: ${NUM_EPOCHS}
SEED: ${SEED}
datasetMounts:
- datasetId: ${WINOGRANDE_DATASET}
containerPath: ${MOUNT_POINT}
- datasetId: ${MODEL_DATASET}
containerPath: ${MODEL_MOUNT_POINT}
requirements:
gpuCount: ${GPU_COUNT}
gpuType: v100
preemptible: true