config_azureblob.yml 1.76 KB
Newer Older
1
2
3
searchSpaceFile: search_space.json
trialCommand: python3 mnist.py
trialGpuNumber: 0
4
trialConcurrency: 1
5
maxTrialNumber: 10
6
tuner:
7
  name: TPE
8
9
  classArgs:
    optimize_mode: maximize
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
trainingService:
  platform: remote
  machineList:
    - host: ${your server's IP or domain name}
      user: ${your user name}
      ssh_key_file: ~/.ssh/id_rsa  # We recommend public key over password, it's more secure and convenient.
    # You can specify more than one SSH servers:
    - host: 123.123.123.123
      port: 10022
      user: nniuser
      password: 12345
      pythonPath: /usr/bin  # Other examples:
                            #   /opt/python3.9/bin
                            #   C:/Python39
                            #   C:/Users/USERNAME/.conda/envs/ENVNAME;C:/Users/USERNAME/.conda/envs/ENVNAME/Scripts;C:/Users/USERNAME/.conda/envs/ENVNAME/Library/bin
25
26
sharedStorage:
  storageType: AzureBlob
J-shang's avatar
J-shang committed
27
28
  # please set localMountPoint as absolute path and localMountPoint should outside the code directory
  # because nni will copy user code to localMountPoint
29
  localMountPoint: ${your/local/mount/point}
J-shang's avatar
J-shang committed
30
31
  # remoteMountPoint is the mount point on training service machine, it can be set as both absolute path and relative path
  # make sure you have `sudo` permission without password on training service machine
32
33
34
35
36
37
38
39
  remoteMountPoint: ${your/remote/mount/point}
  storageAccountName: ${replace_to_your_storageAccountName}
  storageAccountKey: ${replace_to_your_storageAccountKey}
  containerName: ${replace_to_your_containerName}
  # usermount means you have already mount this storage on localMountPoint
  # nnimount means nni will try to mount this storage on localMountPoint
  # nomount means storage will not mount in local machine, will support partial storages in the future 
  localMounted: nnimount