File size: 1,057 Bytes
834675c | 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 | {
"name": "addition_sharp position coupling seed 2",
"task": {
"op": "add",
"max_digits": 20,
"min_digits": 1,
"carry_heavy": 0.0,
"reverse": true,
"zero_pad": true,
"positions": "coupled",
"blanks": 0,
"blanks_fixed": true,
"offset_max": 100,
"test_digits": [
5,
10,
15,
20,
25,
30,
40,
50,
60,
80,
100
],
"test_examples": 256,
"kind": "arithmetic"
},
"model": {
"arch": "transformer",
"d_model": 256,
"n_layers": 4,
"n_heads": 4,
"d_mlp": 1024,
"max_positions": 1024,
"pos_embed": "absolute",
"attn_scale": 2.0,
"dropout": 0.0
},
"train": {
"steps": 50000,
"batch_size": 256,
"lr": 0.0003,
"weight_decay": 0.1,
"betas": [
0.9,
0.98
],
"warmup_steps": 1000,
"cosine": true,
"amp": true,
"eval_every": 2500,
"checkpoint_every": 10000,
"seed": 2,
"wandb": false
}
} |