def test_layers_with_residual(name, kwargs, in_data, out_data):
cfg = {"@layers": "residual.v1", "layer": {"@layers": name, **kwargs}}
filled = registry.fill_config({"config": cfg})
model = registry.make_from_config(filled)["config"]
if "LSTM" in name:
model = with_padded(model)
model.initialize(in_data, out_data)
Y, backprop = model(in_data, is_train=True)
assert_data_match(Y, out_data)
dX = backprop(Y)
assert_data_match(dX, in_data)
def test_layers_from_config(name, kwargs, in_data, out_data):
cfg = {"@layers": name, **kwargs}
filled = registry.fill_config({"config": cfg})
model = registry.make_from_config(filled)["config"]
if "LSTM" in name:
model = with_padded(model)
if "FeatureExtractor" in name: # can't validate fake docs:
DATA_VALIDATION.set(False)
model.initialize(in_data, out_data)
Y, backprop = model(in_data, is_train=True)
assert_data_match(Y, out_data)
dX = backprop(Y)
assert_data_match(dX, in_data)
DATA_VALIDATION.set(True)