19 lines
466 B
YAML
19 lines
466 B
YAML
encoding_scheme: nb
|
|
num_features: 8
|
|
vocab_name: MusicTokenVocabNB
|
|
model_name: NestedMusicTransformer
|
|
input_embedder_name: SummationEmbedder
|
|
main_decoder_name: XtransformerDecoder
|
|
sub_decoder_name: DiffusionDecoder
|
|
model_dropout: 0.2
|
|
input_embedder:
|
|
num_layer: 1
|
|
num_head: 8
|
|
main_decoder:
|
|
dim_model: 768
|
|
num_layer: 16
|
|
num_head: 12
|
|
sub_decoder:
|
|
decout_window_size: 1 # 1 means no previous decoding output added
|
|
num_layer: 1
|
|
feature_enricher_use: False |