make_recipient
Make a recipient model according to donor and surgery plan for shrinking a layer.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
donor |
tf.keras.Model
|
The donor model. |
required |
layer |
str
|
The layer name to shrink. e.g.(hidden, phonology, cleanup in PMSP). |
required |
keep_n |
int
|
The number of units to keep. |
required |
make_model_fn |
Callable
|
A function that make the original and new model. |
required |