File size: 1,690 Bytes
1cae80b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
wandb_version: 1

DATASET:
  desc: null
  value:
    MEAN:
    - 0.5
    - 0.5
    - 0.5
    NAME: IFFI
    ROOT: ../../Downloads/IFFI-dataset/train
    SIZE: 256
    STD:
    - 0.5
    - 0.5
    - 0.5
    TEST_ROOT: ../../Downloads/IFFI-dataset/test
MODEL:
  desc: null
  value:
    D:
      NAME: 1-ChOutputDiscriminator
      NUM_CHANNELS: 32
      NUM_CRITICS: 5
      SOLVER:
        BETAS:
        - 0.5
        - 0.9
        DECAY_RATE: 0.5
        LR: 0.001
        SCHEDULER: []
    IFR:
      DESTYLER_CHANNELS: 32
      NAME: InstaFilterRemovalNetwork
      NUM_CHANNELS: 32
      SOLVER:
        BETAS:
        - 0.5
        - 0.9
        DECAY_RATE: 0
        LR: 0.0002
        SCHEDULER: []
    IS_TRAIN: true
    NAME: ifrnet
    NUM_CLASS: 17
OPTIM:
  desc: null
  value:
    ADVERSARIAL: 0.001
    AUX: 0.5
    GP: 10
    MASK: 1
    RECON: 1.4
    SEMANTIC: 0.0001
    TEXTURE: 0.001
SYSTEM:
  desc: null
  value:
    NUM_GPU: 2
    NUM_WORKERS: 4
TEST:
  desc: null
  value:
    ABLATION: false
    BATCH_SIZE: 64
    IMG_ID: 52
    OUTPUT_DIR: ./outputs
    WEIGHTS: ''
TRAIN:
  desc: null
  value:
    BATCH_SIZE: 8
    IS_TRAIN: true
    LOG_INTERVAL: 100
    NUM_TOTAL_STEP: 120000
    RESUME: true
    SAVE_DIR: ./weights
    SAVE_INTERVAL: 1000
    SHUFFLE: true
    START_STEP: 0
    TUNE: false
    VISUALIZE_INTERVAL: 100
WANDB:
  desc: null
  value:
    ENTITY: vvgl-ozu
    LOG_DIR: ./logs/ifrnet_IFFI_120000step_8bs_0.0002lr_2gpu_9run
    NUM_ROW: 0
    PROJECT_NAME: instagram-filter-removal
    RUN: 9
_wandb:
  desc: null
  value:
    cli_version: 0.9.1
    framework: torch
    is_jupyter_run: false
    is_kaggle_kernel: false
    python_version: 3.6.9