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 create_generator(train_dir, steps_per_beat=4, hparams=None): melody_encoder_decoder = attention_rnn_encoder_decoder.MelodyEncoderDecoder( ) details = generator_pb2.GeneratorDetails( id='attention_rnn', description='Attention RNN Generator') return melody_rnn_sequence_generator.MelodyRnnSequenceGenerator( details, train_dir, melody_encoder_decoder, attention_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 create_generator(checkpoint, bundle, steps_per_quarter=4, hparams=None, generator_id=DEFAULT_ID): melody_encoder_decoder = basic_rnn_encoder_decoder.MelodyEncoderDecoder() details = generator_pb2.GeneratorDetails(id=generator_id, description='Basic RNN Generator') return melody_rnn_sequence_generator.MelodyRnnSequenceGenerator( details, checkpoint, bundle, melody_encoder_decoder, basic_rnn_graph.build_graph, steps_per_quarter, {} if hparams is None else hparams)
def create_generator(checkpoint, bundle, steps_per_quarter=4, hparams=None): melody_encoder_decoder = attention_rnn_encoder_decoder.MelodyEncoderDecoder() details = generator_pb2.GeneratorDetails( id='attention_rnn', description='Attention RNN Generator') return melody_rnn_sequence_generator.MelodyRnnSequenceGenerator( details, checkpoint, bundle, melody_encoder_decoder, attention_rnn_graph.build_graph, steps_per_quarter, {} if hparams is None else hparams)