Masioki commited on
Commit
bd81657
1 Parent(s): e5202c5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ ASR encoder: [Whisper small](https://huggingface.co/openai/whisper-small) encode
37
  Prosody encoder: 2 layer transformer encoder with initial dense projection
38
  Backbone: [Phi 3 mini](https://huggingface.co/microsoft/Phi-3-mini-4k-instruct)
39
  Fusion: 2 residual cross attention fusion layers (F_asr x F_text and F_prosody x F_text) with dense layer on top
40
- Pooling: Weighted mean
41
  Multi-label classification head: 2 dense layers with two dropouts 0.3 and Tanh activation inbetween
42
 
43
 
 
37
  Prosody encoder: 2 layer transformer encoder with initial dense projection
38
  Backbone: [Phi 3 mini](https://huggingface.co/microsoft/Phi-3-mini-4k-instruct)
39
  Fusion: 2 residual cross attention fusion layers (F_asr x F_text and F_prosody x F_text) with dense layer on top
40
+ Pooling: Self-attention
41
  Multi-label classification head: 2 dense layers with two dropouts 0.3 and Tanh activation inbetween
42
 
43