def main(unused_argv):
    melody_rnn_generate.setup_logs()

    with basic_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)
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)
Exemple #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)