search

neuralmagic / sparsezoo

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes
Apache License 2.0
371 stars 25 forks source link

`RegistryMixin` improved alias management #404

Closed dbogunowicz closed 10 months ago

dbogunowicz commented 12 months ago

Feature Description

Adds additional features the the RegistryMixin:

  1. Separates concept of name and alias. Now, we can only have one name and several aliases
  2. Registered names and aliases can be independently inspected using the appropriate methods
  3. Registering and looking up names will be now robust wrt underscores and spaces (see the docstrings and tests)
  4. Improves tests so their are more readable and extendable
dbogunowicz commented 10 months ago

@bfineran @Satrat you are both right, standardising the alias is necessary. Introduced appropriate changes.