NNI 설치와 시작

NNI는 pip으로 설치하고, 실험 단위로 동작해요. 탐색 공간(search space)과 튜닝 알고리즘을 정하고 트라이얼(trial)을 돌리면, 최적 하이퍼파라미터를 찾아주는 구조예요.

출처: microsoft/nni README

NNI 설치 방법을 볼게요. 현재 릴리스를 pip으로 설치하는 게 가장 간단해요.

pip install nni

최신 버전으로 업데이트하려면 --upgrade 플래그를 붙이면 돼요.

pip install --upgrade nni

NNI 라이브러리 구조를 보면, 코어 파이썬 패키지(nni/) 아래에 하이퍼파라미터 튜닝, NAS, 모델 압축 관련 모듈이 나뉘어 있어요. 문서 폴더(docs/)와 예제(examples/)가 함께 있어서, 처음 시작할 땐 예제를 그대로 따라가면 되고요.

트레이닝 서비스도 다양해요. 로컬 머신뿐 아니라 원격 SSH 서버, Azure Machine Learning(AML), Kubernetes 기반 환경에서도 트라이얼을 분산 실행할 수 있어요. HPO 예제부터 PyTorch·TensorFlow 튜토리얼, NAS 벤치마크, 프루닝·양자화 스피드업 예제까지 풍부하게 준비돼 있어요.

NNI 공식 문서 홈(nni.readthedocs.io)에는 설치 가이드, 예제, Python API 레퍼런스가 정리돼 있어서, 자기 모델에 붙일 때 참고하면 좋아요.