IRIS-AUDIO / SELD

7 stars 8 forks source link

Add Sepformer + force_1d_inputs + embeddings #98

Closed daniel03c1 closed 3 years ago

daniel03c1 commented 3 years ago

Sepformer 기반 블럭을 넣어봤습니다. 1d로 shape을 강제하는 것을 어디에 넣을까 하다가 force_1d_inputs라는 함수로 layers.py에 넣었습니다. 이외에도 embedding도 layers에 넣었습니다. 사용방법은 x + embedding(x.shape)(x) 이런식으로 하면 됩니다. 이에 맞춰 transformer_encoder_layer에 조금 변경을 더했습니다.