diff options
| author | Nathan Bailey <nathan.bailey@arm.com> | 2024-04-08 15:42:11 +0100 |
|---|---|---|
| committer | Nathan Bailey <nathan.bailey@arm.com> | 2024-05-07 11:48:11 +0000 |
| commit | 0999ba0f4381ce1e2e8b06a932bfe693692223e2 (patch) | |
| tree | cdc1061b885f9bfa7548b5c9bc569750a7ba957c /tests/test_nn_rewrite_core_train.py | |
| parent | 7565b9a5460ddc7d74eb74dfb6d0c4264f99ebaf (diff) | |
| download | mlia-0999ba0f4381ce1e2e8b06a932bfe693692223e2.tar.gz | |
fix: Fixes MAE discrepancies of rewrites
- Adds model.evaluate stage to retrieve correct MAE
- Sets augmentations to none by default
- Enables MAE calculations using dequantized data (if needed)
Resolves: MLIA-972
Signed-off-by: Nathan Bailey <nathan.bailey@arm.com>
Change-Id: Ia838b2d86caf7b10ad6e4d87bf6aa9e27c80bb72
Diffstat (limited to 'tests/test_nn_rewrite_core_train.py')
0 files changed, 0 insertions, 0 deletions