def from_config(cls, config, custom_objects=None):
if "residual_fn" in config:
config = config.copy()
residual_function = _parse_config_to_function(
config, custom_objects, "residual_fn", "residual_fn_type",
"residual_fn_module")
config["residual_fn"] = residual_function
return super(ResidualWrapper,
cls).from_config(config, custom_objects=custom_objects)
def from_config(cls, config, custom_objects=None):
if "dropout_fn" in config:
config = config.copy()
dropout_state_filter = _parse_config_to_function(
config, custom_objects, "dropout_fn", "dropout_fn_type",
"dropout_fn_module")
config.pop("dropout_fn")
config["dropout_state_filter_visitor"] = dropout_state_filter
return super(DropoutWrapper,
cls).from_config(config, custom_objects=custom_objects)