-
Notifications
You must be signed in to change notification settings - Fork 1
/
sweep.yaml
74 lines (74 loc) · 1.28 KB
/
sweep.yaml
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
program: train.py
method: grid
metric:
goal: maximize
name: best_f1
parameters:
per_device_train_batch_size:
values:
- 16
distribution: categorical
num_train_epochs:
values:
- 10
distribution: categorical
learning_rate:
values:
- 0.00002
distribution: categorical
seed:
values:
- 1
- 2
- 3
- 4
- 5
distribution: categorical
aug_prob:
values:
- 0.07
distribution: categorical
model_name_or_path:
values:
- xlm-roberta-base
distribution: categorical
output_dir:
values:
- /tmp/cbd/
distribution: categorical
dataset_name:
values:
- allegro/klej-cbd
distribution: categorical
max_seq_length:
values:
- 128
distribution: categorical
evaluation_strategy:
values:
- steps
distribution: categorical
eval_steps:
values:
- 50
distribution: categorical
logging_strategy:
values:
- steps
distribution: categorical
logging_steps:
values:
- 50
distribution: categorical
report_to:
values:
- wandb
distribution: categorical
overwrite_output_dir:
values:
- True
distribution: categorical
do_train:
values:
- True
distribution: categorical