"driver/driver.cpp" did not exist on "0a2657312ec62a65e92a36cebd7d3b2a3c0712e1"
Unverified Commit d204d8bf authored by politecat314's avatar politecat314 Committed by GitHub
Browse files

added config_windows.yml for examples in /examples/trials (#4121)

parent e76f196d
searchSpaceFile: search_space.json
trialCommand: python main.py
trialConcurrency: 1
maxTrialNumber: 10
tuner:
name: TPE
classArgs:
optimize_mode: minimize
trainingService: # For other platforms, check mnist-pytorch example
platform: local
searchSpaceFile: search_space.json
trialCommand: python main.py
trialGpuNumber: 1
trialConcurrency: 4
maxTrialNumber: 10
tuner:
name: TPE
classArgs:
optimize_mode: maximize
trainingService: # For other platforms, check mnist-pytorch example
platform: local
maxTrialNumberPerGpu: 2
useActiveGpu: false # NOTE: Use "true" if you are using an OS with graphical interface (e.g. Windows 10, Ubuntu desktop)
# Check the doc for details: https://nni.readthedocs.io/en/latest/reference/experiment_config.html#useactivegpu
searchSpaceFile: search_net.json
trialCodeDirectory: EfficientNet-PyTorch
trialCommand: python main.py /data/imagenet -j 12 -a efficientnet --batch-size 48 --lr 0.048 --wd 1e-5 --epochs 5 --request-from-nni
trialGpuNumber: 1
trialConcurrency: 4
maxTrialNumber: 100
tuner:
className: tuner.FixedProductTuner
codeDirectory: .
classArgs:
product: 2
trainingService: # For other platforms, check mnist-pytorch example
platform: local
useActiveGpu: false # NOTE: Use "true" if you are using an OS with graphical interface (e.g. Windows 10, Ubuntu desktop)
# Check the doc for details: https://nni.readthedocs.io/en/latest/reference/experiment_config.html#useactivegputrial:
trialCommand: python trial.py
trialGpuNumber: 0
trialConcurrency: 1
maxTrialNumber: 10
maxExperimentDuration: 1h
searchSpace: {} # hard-coded in tuner
tuner:
className: customer_tuner.CustomerTuner
codeDirectory: ../../tuners/ga_customer_tuner
classArgs:
optimize_mode: maximize
trainingService: # For other platforms, check mnist-pytorch example
platform: local
useAnnotation: true
trialCommand: python train.py
trialGpuNumber: 0
trialConcurrency: 2
maxTrialNumber: 10
tuner:
name: TPE
classArgs:
optimize_mode: maximize
trainingService: # For other platforms, check mnist-pytorch example
platform: local
useAnnotation: true
trialCommand: python mnist.py
trialGpuNumber: 0
trialConcurrency: 1
maxTrialNumber: 10
maxExperimentDuration: 1h
tuner:
name: TPE
classArgs:
optimize_mode: maximize
trainingService: # For other platforms, check mnist-pytorch example
platform: local
searchSpaceFile: search_space.json
trialCommand: python mnist-keras.py
trialGpuNumber: 0
trialConcurrency: 1
maxTrialNumber: 10
maxExperimentDuration: 1h
tuner:
name: BatchTuner
trainingService: # For other platforms, check mnist-pytorch example
platform: local
authorName: default
experimentName: example_mnist-keras
trialConcurrency: 1
maxExecDuration: 1h
maxTrialNum: 10
#choice: local, remote, pai
trainingServicePlatform: local
searchSpacePath: search_space.json
#choice: true, false
useAnnotation: false
tuner:
#choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner
#SMAC (SMAC should be installed through nnictl)
builtinTunerName: TPE
classArgs:
#choice: maximize, minimize
optimize_mode: maximize
trial:
command: python mnist-keras.py
codeDir: .
gpuNum: 0
searchSpaceFile: search_space.json
trialCommand: python mnist.py
trialGpuNumber: 0
trialConcurrency: 2
maxTrialNumber: 100
maxExperimentDuration: 1h
tuner:
name: TPE
classArgs:
optimize_mode: maximize
trainingService: # For other platforms, check mnist-pytorch example
platform: local
useActiveGpu: false # NOTE: Use "true" if you are using an OS with graphical interface (e.g. Windows 10, Ubuntu desktop)
# Check the doc for details: https://nni.readthedocs.io/en/latest/reference/experiment_config.html#useactivegpu
searchSpaceFile: search_space.json
trialCommand: python mnist.py
trialGpuNumber: 1
trialConcurrency: 3
maxTrialNumber: 100
maxExperimentDuration: 2h
tuner:
name: PBTTuner
classArgs:
optimize_mode: maximize
trainingService: # For other platforms, check mnist-pytorch example
platform: local
useActiveGpu: false # NOTE: Use "true" if you are using an OS with graphical interface (e.g. Windows 10, Ubuntu desktop)
# Check the doc for details: https://nni.readthedocs.io/en/latest/reference/experiment_config.html#useactivegpu
searchSpaceFile: search_space.json
trialCommand: python mnist.py
trialGpuNumber: 0
trialConcurrency: 1
tuner:
name: TPE
classArgs:
optimize_mode: maximize
trainingService:
platform: local
searchSpaceFile: search_space.json
trialCommand: python mnist.py
trialGpuNumber: 0
trialConcurrency: 1
tuner:
name: TPE
classArgs:
optimize_mode: maximize
trainingService:
platform: local
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment