max_seq_len=256,
z_size=512,
enc_rnn_size=[2048, 2048],
dec_rnn_size=[1024, 1024],
free_bits=256,
max_beta=0.2,
)),
note_sequence_augmenter=None,
data_converter=trio_16bar_converter,
train_examples_path=None,
eval_examples_path=None,
)
CONFIG_MAP['hier-trio_16bar'] = Config(
model=MusicVAE(
lstm_models.HierarchicalLstmEncoder(
lstm_models.BidirectionalLstmEncoder, [16, 16]),
lstm_models.HierarchicalLstmDecoder(
lstm_models.SplitMultiOutLstmDecoder(
core_decoders=[
lstm_models.CategoricalLstmDecoder(),
lstm_models.CategoricalLstmDecoder(),
lstm_models.CategoricalLstmDecoder()
],
output_depths=[
90, # melody
90, # bass
512, # drums
]),
level_lengths=[16, 16],
disable_autoregression=True)),
hparams=merge_hparams(
max_seq_len=256,
z_size=512,
enc_rnn_size=[2048, 2048],
dec_rnn_size=[1024, 1024],
free_bits=256,
max_beta=0.2,
)),
note_sequence_augmenter=None,
data_converter=trio_16bar_converter,
train_examples_path=None,
eval_examples_path=None,
)
CONFIG_MAP['hier-trio_16bar'] = Config(
model=MusicVAE(
lstm_models.HierarchicalLstmEncoder(
lstm_models.BidirectionalLstmEncoder, [16, 16]),
lstm_models.HierarchicalLstmDecoder(
lstm_models.SplitMultiOutLstmDecoder(
core_decoders=[
lstm_models.CategoricalLstmDecoder(),
lstm_models.CategoricalLstmDecoder(),
lstm_models.CategoricalLstmDecoder()
],
output_depths=[
90, # melody
90, # bass
512, # drums
]),
level_lengths=[16, 16],
disable_autoregression=True)),
hparams=merge_hparams(
