Reference_based_Skectch_Image_Colorization/config.yml at master · Jungjaewon/Reference_based_Skectch_Image_Colorization · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

MODEL_CONFIG:
  IMG_SIZE: 256

TRAINING_CONFIG:

  EPOCH: 100
  BATCH_SIZE: 8

  # lr and lambda
  G_LR : 10e-4
  D_LR : 20e-4
  LAMBDA_TR  : 12.0
  LAMBDA_G_FAKE  : 1
  LAMBDA_G_RECON : 30
  LAMBDA_G_SYTLE : 50
  LAMBDA_G_PERCEP : 0.01
  LAMBDA_D_FAKE  : 1
  LAMBDA_D_REAL  : 1
  LAMBDA_GP : 20000

  # Data augmentation
  AUGMENT : 'tps' # or 'tps, elastic

  #Triplet loss
  TRIPLE_LOSS : 'False' # 'True' or 'False'
  GAN_LOSS : 'lsgan'

  # Optimizer
  OPTIM : ADAM
  BETA1: 0.5
  BETA2: 0.999
  D_CRITIC : 1
  G_CRITIC : 1

  # Settings
  NUM_WORKER : 4
  MODE : 'train'
  CPU_SEED : 123
  GPU_SEED : 456
  USE_TENSORBOARD : 'False' # 'True' or 'False'

  # noise parameter
  DIST : 'uniform' # of 'gaussian'
  A : -1
  B : 1
  MEAN : 0
  STD : 0.01
  #spectral norm
  G_SPEC : 'False'
  D_SPEC : 'True'

  # Directory
  IMG_DIR : 'data'
  TRAIN_DIR : 'colorization_gan4'
  LOG_DIR : 'logs'
  SAMPLE_DIR : 'samples'
  RESULT_DIR : 'results'
  MODEL_DIR : 'models'

  # GPU
  GPU: 1

  # Step Size

  SAMPLE_STEP : 1 # epoch based
  LOG_STEP : 100 # iteration based
  SAVE_STEP : 5 # epoch based
  SAVE_START : 10
  LR_DECAY_STEP : 20

TEST_CONFIG:
  TEST_EPOCH : 100