File size: 506 Bytes
62a2f1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
{
"batch_size": 64,
"encode": "one-hot",
"epochs": 100,
"early_stop": 20,
"lr": 0.001,
"convolution_layers": {
"n_layers": 4,
"filters": [1024, 512, 256, 128],
"kernel_sizes": [8, 16, 32, 64]
},
"transformer_layers": {
"n_layers": 0,
"attn_key_dim": [16, 16, 16],
"attn_heads": [2048, 2048, 2048]
},
"n_dense_layer": 1,
"dense_neurons1": 64,
"dropout_conv": "yes",
"dropout_prob": 0.4,
"pad": "same"
}
|