File size: 1,074 Bytes
62ad7b9 74a6211 7c814dc 1e90a21 |
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 |
{
"model_type": "custom",
"custom_model": {
"module": "path.to.your.custom_module",
"class": "CSUMLMModel",
"config": {
"text_encoder": {
"model_name": "bert-base-uncased",
"pretrained_weights": "path/to/bert-base-uncased"
},
"image_encoder": {
"model_name": "resnet50",
"pretrained_weights": "path/to/resnet50"
},
"audio_encoder": {
"model_name": "wav2vec2-base",
"pretrained_weights": "path/to/wav2vec2-base"
},
"fusion_layer": {
"type": "concatenate",
"output_dim": 1024
},
"output_layer": {
"type": "dense",
"units": 1,
"activation": "sigmoid"
},
"attention_mechanism": {
"traditional_attention": {
"type": "bahdanau"
},
"self_attention": {
"type": "scaled_dot_product"
},
"linear_attention": {
"type": "linear"
}
},
"belief_desire_intent_tree": {
"type": "recursive_neural_network"
}
}
}
} |