Esempio n. 1
0
def main(unused_argv):
    melody_rnn_generate.setup_logs()

    with lookback_rnn_generator.create_generator(
            melody_rnn_generate.get_checkpoint(),
            melody_rnn_generate.get_bundle(),
            melody_rnn_generate.get_steps_per_quarter(),
            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)
Esempio n. 2
0
def main(unused_argv):
  melody_rnn_generate.setup_logs()

  with lookback_rnn_generator.create_generator(
      melody_rnn_generate.get_checkpoint(),
      melody_rnn_generate.get_bundle(),
      melody_rnn_generate.get_steps_per_quarter(),
      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)
Esempio n. 3
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)