Skip to content

make_recipient

Source code

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