def __init__(self,
generator_name,
num_bars_to_generate,
hparams,
checkpoint=None,
bundle_file=None):
self._num_bars_to_generate = num_bars_to_generate
if not checkpoint and not bundle_file:
raise GeneratorException(
'No generator checkpoint or bundle location supplied.')
if (checkpoint or generator_name or hparams) and bundle_file:
raise GeneratorException(
'Cannot specify both bundle file and checkpoint, generator_name, '
'or hparams.')
bundle = None
if bundle_file:
bundle = sequence_generator_bundle.read_bundle_file(bundle_file)
generator_name = bundle.generator_details.id
if generator_name not in _GENERATOR_FACTORY_MAP:
raise GeneratorException('Invalid generator name given: %s',
generator_name)
generator = _GENERATOR_FACTORY_MAP[generator_name].create_generator(
checkpoint=checkpoint, bundle=bundle, hparams=hparams)
generator.initialize()
self._generator = generator
def get_bundle():
"""Returns a generator_pb2.GeneratorBundle object based read from bundle_file.
Returns:
Either a generator_pb2.GeneratorBundle or None if the bundle_file flag is
not set or the save_generator_bundle flag is set.
"""
if should_save_generator_bundle():
return None
bundle_file = get_bundle_file()
if bundle_file is None:
return None
return sequence_generator_bundle.read_bundle_file(bundle_file)