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(