前言
使用CPU调参的例子就略过了,网络上教程很多,这篇博客主要针对GPU环境下NNI配置问题。
配置
使用GPU需要注意修改config.myl的几个地方:
config.myl
authorName: default
experimentName: titanic
trialConcurrency: 1
maxExecDuration: 1h
maxTrialNum: 100
#choice: local, remote
trainingServicePlatform: local
searchSpacePath: search_space.json
#choice: true, false
useAnnotation: false
tuner:
#choice: TPE, Random, Anneal, Evolution
builtinTunerName: TPE
classArgs:
#choice: maximize, minimize
optimize_mode: maximize
trial:
command: python train.py
codeDir: .
gpuNum: 1
localConfig:
useActiveGpu: true
maxTrialNumPerGpu: 5
gpuIndices: 1
</