Fix example issue (#4456)

microsoft · Jan 10, 2022 · 452e69f · 452e69f
1 parent 100b95f
commit 452e69f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 52 deletions.
diff --git a/examples/__init__.py b/examples/__init__.py
diff --git a/examples/experiment_config.yml b/examples/experiment_config.yml
diff --git a/examples/trials/mnist-pytorch/config_assessor.yml b/examples/trials/mnist-pytorch/config_assessor.yml
@@ -1,27 +1,16 @@
-authorName: default
-experimentName: example_mnist_pytorch
-trialConcurrency: 1
-maxExecDuration: 1h
-maxTrialNum: 50
-#choice: local, remote
-trainingServicePlatform: local
-searchSpacePath: search_space.json
-#choice: true, false
-useAnnotation: false
+searchSpaceFile: search_space.json
+trialCommand: python3 mnist.py  # NOTE: change "python3" to "python" if you are using Windows
+trialGpuNumber: 0
+trialConcurrency: 4
+maxTrialNumber: 20
 tuner:
-  #choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner, GPTuner
-  #SMAC (SMAC should be installed through nnictl)
-  builtinTunerName: TPE
+  name: TPE
   classArgs:
-    #choice: maximize, minimize
     optimize_mode: maximize
 assessor:
-  #choice: Medianstop, Curvefitting
-  builtinAssessorName: Curvefitting
+  name: Curvefitting
   classArgs:
     epoch_num: 20
     threshold: 0.9
-trial:
-  command: python3 mnist.py
-  codeDir: .
-  gpuNum: 0
+trainingService:
+  platform: local