AnalogAINAS: A modular and extensible Analog-aware Neural Architecture Search (NAS) library.
Project description
analogai-nas
AnalogAINas is a modular and flexible framework to facilitate implementation of Analog-aware Neural Architecture Search. It offers high-level classes to define: the search space, the accuracy evaluator, and the search strategy. It leverages the aihwkit framework to apply hardware-aware training with analog non-idealities and noise included. AnalogAINAS obtained architectures are more robust during inference on Analog Hardware. We also include two evaluators trained to rank the architectures according to their analog training accuracy.
Setup
While installing the repository, creating a new conda environment is recomended.
git clone https://github.com/IBM/analog-nas/
pip install -r requirements.txt
pip setup.py install
Usage
To get started, check out nas_search_demo.py to make sure that the installation went well.
This python script describes how to use the package.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for analogainas-0.1.0-py2.py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 070bd0264a6a844ececb59f87ec9352e613e72af11f6001b7eae6bfb7bf29b98 |
|
| MD5 | 38ffd1784519810dd4a9bd4433eb2d7c |
|
| BLAKE2b-256 | db8bf63bd59e27835050d68fb0b0221f2583bd07819732fb08ad4c7361b8ad06 |
