def create_generator(checkpoint, steps_per_beat=4, hparams=None):
melody_encoder_decoder = basic_rnn_encoder_decoder.MelodyEncoderDecoder()
details = generator_pb2.GeneratorDetails(id='basic_rnn',
description='Basic RNN Generator')
return melody_rnn_sequence_generator.MelodyRnnSequenceGenerator(
details, checkpoint, melody_encoder_decoder,
basic_rnn_graph.build_graph, steps_per_beat,
{} if hparams is None else hparams)
def main(unused_argv):
melody_encoder_decoder = basic_rnn_encoder_decoder.MelodyEncoderDecoder()
details = generator_pb2.GeneratorDetails(id='basic_rnn',
description='Basic RNN Generator')
with melody_rnn_sequence_generator.MelodyRnnSequenceGenerator(
details, melody_rnn_generate.get_train_dir(),
melody_encoder_decoder, basic_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 = basic_rnn_encoder_decoder.MelodyEncoderDecoder()
melody_rnn_generate.run(melody_encoder_decoder,
basic_rnn_graph.build_graph)
def main(unused_argv): melody_encoder_decoder = basic_rnn_encoder_decoder.MelodyEncoderDecoder() melody_rnn_create_dataset.run(melody_encoder_decoder)
