Skip to content

Prune PAF graphs before evaluation#2813

Merged
n-poulsen merged 4 commits intopytorch_dlcfrom
niels/prune_paf_graphs
Dec 13, 2024
Merged

Prune PAF graphs before evaluation#2813
n-poulsen merged 4 commits intopytorch_dlcfrom
niels/prune_paf_graphs

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
@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.

2 participants