Microsoft AutoML Tool - NNI



마이크로소프트 Github 에서 제공중인 Neural Network Intelligence는 AutoMl의 하이퍼파라미터 튜닝, 모델 검색, 전처리 튜닝 등을 손쉽게 Wrapping으로 구현하여 WebUI로 모니터링 할수있는 툴이다.


위의 그림에서와 같이 먼저 기존 학습코드 (train.py)에 nni 코드를 추가한뒤(뒤에 예제있음) nnictl 커맨트를 이용하여 실행하면 Web UI로 모니터링 가능하다

ofiicial document가 잘 작성되어있기때문에
https://nni.readthedocs.io/en/latest/Tutorial/QuickStart.html
퀵스타트문서를 통해 진행하는 것을 추천합니다.




------------------아래에는 mac환경, pyenv의 가상파이썬환경에서 예제를 실행하는 방법입니다.

작업환경은 mac10.14 이며 pyenv 를 이용하여 진행합니다. (파이썬 가상환경 (예시에서는 가상환경이름을 pyenv_example로 합니다.))

가상환경 실행
$pyenv activate pyenv_example

nni 설치
$python3 -m pip install --upgrade nni

예제 소스코드 다운로드
[tensorflow로 구현되어있는 mnist예제를 사용합니다.]
$git clone -b v1.4 htttps://github.com/Microsoft/nni.git

예제 실행
[~/.pyenv/versions/pyenv_example/bin 는 가상환경의 python 이 설치된 경로입니다.]
$pip install tensorflow==2.0.0
$~/.pyenv/versions/pyenv_example/bin/nnictl create --config nni/example/trials/mnist-tfv2.config.yml

댓글