Skip to content

Conversation

@n-poulsen
Copy link
Contributor

Cleans the method to prune PAF graphs. For models with Part-Affinity Heads (DLCRNetHead), calls the method to prune the PAF graph before evaluating the model.

@n-poulsen n-poulsen added DLC3.0🔥 enhancement New feature or request labels Dec 12, 2024
@n-poulsen n-poulsen merged commit 431bd7e into pytorch_dlc Dec 13, 2024
1 check passed
@n-poulsen n-poulsen deleted the niels/prune_paf_graphs branch December 13, 2024 09:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

DLC3.0🔥 enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants