def from_config(cls, config):
return cls(
Container.from_config(config['generator']),
Container.from_config(config['discriminator']),
config['z_shape'],
config['real_shape'],
)
def from_config(cls, config):
return cls(
Container.from_config(config['generator']),
Container.from_config(config['discriminator']),
config['z_shape'],
config['real_shape'],
)
def from_config(cls, config, custom_objects=None):
step_model = Container.from_config(config, custom_objects)
config.pop('name')
config.pop('layers')
config.pop('input_layers')
config.pop('output_layers')
return cls(_step_model=step_model, **config)
