YOLO11 got stuck during pruning while generating pruner

[TenMinutes209]

When I tried to prune an YOLO11 model with Backbone replaced with ShuffleNetV2, I got problems below:

The code stuck in the sentence
pruner = tp.pruner.GroupNormPruner(
model.model,
example_inputs,
importance=tp.importance.GroupNormImportance(), # L2 norm pruning,
iterative_steps=1,
pruning_ratio=pruning_ratio,
ignored_layers=ignored_layers,
unwrapped_parameters=unwrapped_parameters
)

How can I solve this problem? Thank you

[TenMinutes209]

I tried that when I keep the param.requires_grad = False, the code could run but the model wasn't pruned. When I turned param.requires_grad = True, the code stuck.