Skip to content

Define you own pruning method #5

Description

@SalemAmeen

Could you please provide an example of how to use other pruning methods?

I tried to use RandomFilterPruning() but I kept receiving an error

for example
r = RandomFilterPruning()
TypeError: init() missing 4 required positional arguments: 'removal_factor', 'model_compile_fn', 'model_finetune_fn', and 'nb_finetune_epochs'

g = RandomFilterPruning.run_pruning(compile_model,
finetune_model,
nb_finetune_epochs=1,
maximum_pruning_percent=0.05,
maximum_prune_iterations=2,
nb_trained_for_epochs=FIRST_TRAIN_EPOCHS )
TypeError: run_pruning() got an unexpected keyword argument 'nb_finetune_epochs'

Many thanks

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions