diff options
author | Annie Tallund <annie.tallund@arm.com> | 2023-09-19 16:34:14 +0200 |
---|---|---|
committer | Annie Tallund <annie.tallund@arm.com> | 2023-09-26 14:24:39 +0200 |
commit | 8dec455c467b8019223a40e107378845e1419f5d (patch) | |
tree | 823051d364c78e3ce7f70023eb94021d8b051274 /tox.ini | |
parent | ba251631768f25b840e93ece6a4af3db119e6dd1 (diff) | |
download | mlia-8dec455c467b8019223a40e107378845e1419f5d.tar.gz |
MLIA-469 Support batch size > 1 for optimizations
- Add a PruningPolicy to skip layers that are not
supported by the Keras pruning API
- Make dataset generation more generic to support
use-cases beyond classification
Signed-off-by: Annie Tallund <annie.tallund@arm.com>
Change-Id: I198dae2b53860f449f2fdbc71575babceed1ffcf
Diffstat (limited to 'tox.ini')
0 files changed, 0 insertions, 0 deletions