-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathconfig_lip.yaml
50 lines (42 loc) · 1.46 KB
/
config_lip.yaml
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
NAME: 'LIP'
IS_TRAIN: false
VAL:
MASKDIR: '/home/dalong/data/lip/val_covers/'
OUTDIR: '/home/dalong/data/toolbox_pytorch/examples/inpainting/results/lip'
INIT: 'snapshot/LIP_1527670923/G_70_0.pkl'
DATASET:
TRAINDIR: '/home/dalong/data/lip/train_images/'
VALDIR: '/home/dalong/data/lip/val_images/'
INPUT_RES: 256
LOCAL_RES: 128
HOLE_MIN: 32
HOLE_MAX: 127
MEAN: [0.4560, 0.4472, 0.4155]
LOSS:
ALPHA: 0.0004 # 0.0004
TRAIN_G_EPOCHES: -1
TRAIN_D_EPOCHES: -1
# INIT_G: '/home/dalong/data/toolbox_pytorch/examples/inpainting/completionnet_places2.t7'
# INIT_D: ~
INIT_G: '/home/dalong/data/toolbox_pytorch/examples/inpainting/snapshot/LIP_1527649602/G_7_0.pkl'
INIT_D: '/home/dalong/data/toolbox_pytorch/examples/inpainting/snapshot/LIP_1527649602/D_7_0.pkl'
SOLVER:
WORKERS: 4
IMG_PER_GPU: 20
GPU_IDS: [0] # real BATCHSIZE is : IMG_PER_GPU * len(GPU_IDS)
LR: 0.0001
LR_SCHEDULE: 'standard'
OPTIMIZER: 'ADAM' # ADAM | SGD
MOMENTUM: 0.9
WEIGHTDECAY: 0.0004
LOGS:
PRINT_FREQ: 10
SNAPSHOT_FREQ: 99999
SNAPSHOT_MAXFILES: NOTIMPLEMENT
SNAPSHOT_DIR: 'snapshot'
LOG_FREQ: 100
LOG_DIR: 'logs'
LOG_SHUTIL_ON: true
# shutil.copytree('./', './logs2', ignore=shutil.ignore_patterns('./logs','*.zip','*.mp4','*.jpg','*.png','__pycache__', '*.pyc'))
LOG_SHUTIL_IGNORELIST: ['*.zip','*.mp4','*.jpg','*.png','*.pyc','*.so','*.pkl','*.caffemodel', '__pycache__','snapshot','logs']
LOG_SHUTIL_IGNOREFILE: '.gitignore'