示例#1
0
def create_generator(checkpoint, steps_per_beat=4, hparams=None):
    melody_encoder_decoder = attention_rnn_encoder_decoder.MelodyEncoderDecoder(
    )
    details = generator_pb2.GeneratorDetails(
        id='attention_rnn', description='Attention RNN Generator')
    return melody_rnn_sequence_generator.MelodyRnnSequenceGenerator(
        details, checkpoint, melody_encoder_decoder,
        attention_rnn_graph.build_graph, steps_per_beat,
        {} if hparams is None else hparams)
def main(unused_argv):
    melody_encoder_decoder = attention_rnn_encoder_decoder.MelodyEncoderDecoder(
    )
    details = generator_pb2.GeneratorDetails(
        id='attention_rnn', description='Attention RNN Generator')
    with melody_rnn_sequence_generator.MelodyRnnSequenceGenerator(
            details, melody_rnn_generate.get_train_dir(),
            melody_encoder_decoder, attention_rnn_graph.build_graph,
            melody_rnn_generate.get_steps_per_beat(),
            melody_rnn_generate.get_hparams()) as generator:
        melody_rnn_generate.run_with_flags(generator)
def main(unused_argv):
    melody_encoder_decoder = attention_rnn_encoder_decoder.MelodyEncoderDecoder(
    )
    melody_rnn_generate.run(melody_encoder_decoder,
                            attention_rnn_graph.build_graph)
示例#4
0
def main(unused_argv):
    melody_encoder_decoder = attention_rnn_encoder_decoder.MelodyEncoderDecoder(
    )
    melody_rnn_create_dataset.run(melody_encoder_decoder)