Skip to content

Commit 7a1ef99

Browse files
committed
add first wandb sweep config
1 parent 51f8fca commit 7a1ef99

File tree

1 file changed

+35
-0
lines changed

1 file changed

+35
-0
lines changed

wandb_sweep.yaml

+35
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
program: platalea\experiments\flickr8k\transformer.py
2+
method: grid
3+
metric:
4+
goal: "hello world wrt wandb sweep function"
5+
name: "sweep hello world"
6+
parameters:
7+
cyclic_lr_max:
8+
values: [0.001, 0.005, 0.0005]
9+
cyclic_lr_min:
10+
value: 0.0005
11+
batch_size:
12+
value: 32
13+
trafo_d_model:
14+
value: 512
15+
trafo_encoder_layers:
16+
value: 4
17+
trafo_heads:
18+
value: 8
19+
trafo_feedforward_dim:
20+
value: 1024
21+
trafo_dropout:
22+
value: 0
23+
epochs:
24+
value: 27
25+
flickr8k_root:
26+
value: "../flickr1d/flickr1d"
27+
flickr8k_meta:
28+
value: "dataset.json"
29+
audio_features_fn:
30+
value: "mfcc_features.pt"
31+
early_terminate:
32+
type: hyperband
33+
s: 2
34+
eta: 3
35+
max_iter: 27

0 commit comments

Comments
 (0)