Ejemplo n.º 1
0
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)
Ejemplo n.º 2
0
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)
Ejemplo n.º 3
0
def main(unused_argv):
    melody_rnn_generate.setup_logs()

    with attention_rnn_generator.create_generator(
            melody_rnn_generate.get_checkpoint(),
            melody_rnn_generate.get_bundle(),
            melody_rnn_generate.get_steps_per_beat(),
            melody_rnn_generate.get_hparams()) as generator:
        if melody_rnn_generate.should_save_generator_bundle():
            tf.logging.info('Saving generator bundle to %s' %
                            (melody_rnn_generate.get_bundle_file()))
            generator.create_bundle_file(melody_rnn_generate.get_bundle_file())
        else:
            melody_rnn_generate.run_with_flags(generator)
def main(unused_argv):
    with attention_rnn_generator.create_generator(
            melody_rnn_generate.get_train_dir(),
            melody_rnn_generate.get_steps_per_beat(),
            melody_rnn_generate.get_hparams()) as generator:
        melody_rnn_generate.run_with_flags(generator)
Ejemplo n.º 5
0
def main(unused_argv):
  with basic_rnn_generator.create_generator(
      melody_rnn_generate.get_train_dir(),
      melody_rnn_generate.get_steps_per_beat(),
      melody_rnn_generate.get_hparams()) as generator:
    melody_rnn_generate.run_with_flags(generator)
Ejemplo n.º 6
0
def main(unused_argv):
  with basic_rnn_generator.create_generator(
      melody_rnn_generate.get_checkpoint(),
      melody_rnn_generate.get_steps_per_beat(),
      melody_rnn_generate.get_hparams()) as generator:
    melody_rnn_generate.run_with_flags(generator)
Ejemplo n.º 7
0
def main(unused_argv):
  with lookback_rnn_generator.create_generator(
      melody_rnn_generate.get_checkpoint(),
      melody_rnn_generate.get_steps_per_beat(),
      melody_rnn_generate.get_hparams()) as generator:
    melody_rnn_generate.run_with_flags(generator)