19 lines
450 B
YAML
19 lines
450 B
YAML
encoding_scheme: nb
|
|
num_features: 5
|
|
vocab_name: MusicTokenVocabNB
|
|
model_name: AmadeusModel
|
|
input_embedder_name: SummationEmbedder
|
|
main_decoder_name: XtransformerDecoder
|
|
sub_decoder_name: CrossAttention
|
|
model_dropout: 0.1
|
|
input_embedder:
|
|
num_layer: 1
|
|
num_head: 8
|
|
main_decoder:
|
|
dim_model: 512
|
|
num_layer: 6
|
|
num_head: 8
|
|
sub_decoder:
|
|
decout_window_size: 1 # 1 means no previous decoding output added
|
|
num_layer: 1
|
|
feature_enricher_use: True |