Skip to content

MDMM tests and hardware-aware metric functions#37

Open
ArghyaRanjanDas wants to merge 2 commits intocern-nextgen:mainfrom
ArghyaRanjanDas:dev
Open

MDMM tests and hardware-aware metric functions#37
ArghyaRanjanDas wants to merge 2 commits intocern-nextgen:mainfrom
ArghyaRanjanDas:dev

Conversation

@ArghyaRanjanDas
Copy link
Copy Markdown
Contributor

Adds tests for the existing MDMM pruning method and integrates the hardware-aware metric functions (FPGAAwareSparsityMetric and PACAPatternMetric) that were lost during intermediate branch merging.

Also fixes a float32/int32 dtype bug in get_layer_sparsity on the TensorFlow backend.

Supported backends

  • KERAS_BACKEND=tensorflow
  • KERAS_BACKEND=torch

Test files

  • pytest tests/test_mdmm.py tests/test_mdmm_metrics.py

Tests cover metric functions, constraint functions, training phases,
mask correctness, penalty numerical verification, constraint type
variations, L0 mode variations, and edge cases.
@nastiapetrovych
Copy link
Copy Markdown
Collaborator

We can't merge directly to main branch unless it was merged to a dev one @ArghyaRanjanDas

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants