aai-institute / nnbench

A small framework for benchmarking machine learning models.
https://aai-institute.github.io/nnbench/
Apache License 2.0
10 stars 3 forks source link

Add transform submodule, parameter compression transform #124

Closed nicholasjng closed 6 months ago

nicholasjng commented 6 months ago

This is the better way of compressing parameters compared to directly in the benchmark runner, which steals responsibility of the transform that we just introduced.

Refactors nnbench.io.transform->nnbench.transforms, the latter being its own submodule. This is useful to have when adding new builtin transforms, so that they do not have to go into a single file.