Caso de uso del marco de aprendizaje automático automático nni

nni es un marco de aprendizaje automático automático de código abierto de Microsoft

Puede hacer que el ajuste del aprendizaje automático sea más conveniente

La instalación es muy simple, solo usa pip

pip install nni

Esta vez estoy usando la última versión, 2.0

Pero no se puede utilizar en este momento, porque aparecerá:

https://github.com/microsoft/nni/issues/3276

La solución también es muy simple, cambia directamente ruamel.yaml a ruamel_yaml

Edite y modifique el contenido en la ruta a continuación

vim /opt/AN/lib/python3.7/site-packages/nni/tools/nnictl/common_utils.py
vim /opt/AN/lib/python3.7/site-packages/nni/experiment/config/base.py
vim /opt/AN/lib/python3.7/site-packages/nni/tools/package_utils/__init__.py

Entonces se puede usar normalmente

root@lizong:/opt# nnictl --help
usage: nnictl [-h] [--version]
              {ss_gen,create,resume,view,update,stop,trial,experiment,platform,webui,config,log,algo,package,tensorboard,top}
              ...

use nnictl command to control nni experiments

positional arguments:
  {ss_gen,create,resume,view,update,stop,trial,experiment,platform,webui,config,log,algo,package,tensorboard,top}
    ss_gen              automatically generate search space file from trial
                        code
    create              create a new experiment
    resume              resume a new experiment
    view                view a stopped experiment
    update              update the experiment
    stop                stop the experiment
    trial               get trial information
    experiment          get experiment information
    platform            get platform information
    webui               get web ui information
    config              get config information
    log                 get log information
    algo                control nni builtin tuner, assessor and advisor
                        algorithms
    package             this argument is replaced by algo
    tensorboard         manage tensorboard
    top                 monitor the experiment

optional arguments:
  -h, --help            show this help message and exit
  --version, -v

A continuación, ejecute el ejemplo oficial:

nnictl create --config nni/examples/trials/mnist-pytorch/config.yml

Nota: se requiere git clone antes de ejecutar

git clone -b v2.0 https://github.com.cnpmjs.org/Microsoft/nni.git

Después de ejecutar correctamente, puede observar el resultado de la ejecución en la página web:

Dirección de acceso específica ------ ejecución ip de la máquina: 8080

 

Podemos ver que la imagen de arriba se ha estado ejecutando durante 48m54s, y la mejor puntuación es de 99 puntos.

 Mira la curva de hiperparámetros:

Aparecerá la introducción de la operación: