File size: 918 Bytes
f15fde5 b88c8eb f15fde5 b88c8eb f15fde5 b88c8eb f15fde5 |
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 |
{
"acti_name": "gelu",
"architectures": [
"CoNeTTEModel"
],
"beam_size": 3,
"betas": [
0.9,
0.999
],
"d_model": 256,
"decoder_dropout_p": 0.2,
"dim_feedforward": 2048,
"eps": 1e-08,
"gen_test_cands": "generate_expt",
"gen_val_cands": "generate_expt",
"label_smoothing": 0.2,
"lr": 0.0005,
"max_pred_size": 20,
"min_pred_size": 3,
"mixup_alpha": 0.4,
"nhead": 8,
"num_decoder_layers": 6,
"optim_name": "AdamW",
"proj_name": "lin768",
"sched_freq": 1,
"sched_interval": "epoch",
"sched_n_steps": 400,
"sched_name": "cos_decay",
"task_mode": "ds_src",
"task_names": [
"clotho",
"audiocaps",
"macs",
"wavcaps_audioset_sl",
"wavcaps_bbc_sound_effects",
"wavcaps_freesound",
"wavcaps_soundbible"
],
"torch_dtype": "float32",
"transformers_version": "4.29.1",
"use_custom_wd": true,
"verbose": 0,
"weight_decay": 2.0
}
|