augment_layers
