Exemplo n.º 1
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)
Exemplo n.º 3
0
def main(unused_argv):
  with attention_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)