20 lines
467 B
YAML
20 lines
467 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.1
|
|
input_embedder:
|
|
num_layer: 1
|
|
num_head: 8
|
|
main_decoder:
|
|
dim_model: 512
|
|
num_layer: 12
|
|
num_head: 16
|
|
sub_decoder:
|
|
decout_window_size: 1 # 1 means no previous decoding output added
|
|
num_layer: 1
|
|
feature_enricher_use: True
|
|
|