Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "1b467c5770c640ab03a47f38e2f759739d1261e0"
Switch branch/tag
Find file Normal viewHistoryPermalink
config.yml 1.01 KB
Newer Older
qianyj's avatar
create branch for v2.9  
qianyj committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47 
experimentName: example_pix2pix 
searchSpace:
  ngf:
    _type: choice
    _value: [16, 32, 64, 128]
  ndf:
    _type: choice
    _value: [16, 32, 64, 128]
  netG:
    _type: choice
    _value: ["unet_256", "resnet_9blocks"]
  netD:
    _type: choice
    _value: ["basic", "pixel", "n_layers"]
  norm:
    _type: choice
    _value: ["batch", "instance", "none"]
  init_type:
    _type: choice
    _value: ["xavier", "normal", "kaiming", "orthogonal"]
  lr:
    _type: choice
    _value: [0.0001, 0.0002, 0.0005, 0.001, 0.005, 0.01, 0.1]
  beta1:
    _type: uniform
    _value: [0, 1]
  lr_policy:
    _type: choice
    _value: ["linear", "step", "plateau", "cosine"]
  gan_mode:
    _type: choice
    _value: ["vanilla", "lsgan", "wgangp"]
  lambda_L1:
    _type: choice
    _value: [1, 5, 10, 100, 250, 500]
trainingService:
  platform: local
  useActiveGpu: true
  gpuIndices: '0'
trialCodeDirectory: .
trialCommand: python3 pix2pix.py
trialConcurrency: 1
trialGpuNumber: 1
tuner:
  name: TPE
  classArgs:
    optimize_mode: minimize