38 lines
851 B
Plaintext
38 lines
851 B
Plaintext
train.py 2020-12-21 15:03:17
|
|
------------ Options -------------
|
|
batch_size: 16
|
|
beta1: 0.5
|
|
checkpoint_dir: checkpoints/default
|
|
contextual_loss_direction: forward
|
|
data_root: /home/zhanbo/remote/video/video_pair3/
|
|
data_root_imagenet: /home/zhanbo/remote/imagenet_ref_pair/
|
|
domain_invariant: False
|
|
epoch: 40
|
|
gpu_ids: 0,1,2,3
|
|
ic: 7
|
|
image_size: [216, 384]
|
|
load_pretrained_model: True
|
|
lr: 1e-05
|
|
lr_gamma: 0.1
|
|
lr_step: 100
|
|
luminance_noise: 2.0
|
|
nonzero_placeholder_probability: 0.0
|
|
permute_data: True
|
|
print_step: 2
|
|
real_reference_probability: 0.7
|
|
resume: False
|
|
resume_epoch: 0
|
|
tb_log_step: 50
|
|
weight_consistent: 0.02
|
|
weight_contextual: 0.2
|
|
weight_gan: 0.2
|
|
weight_nonlocal_consistent: 0.0
|
|
weight_nonlocal_smoothness: 0.0
|
|
weight_perceptual: 0.001
|
|
weight_smoothness: 5.0
|
|
weigth_l1: 2.0
|
|
with_bad: True
|
|
with_mid: True
|
|
workers: 8
|
|
-------------- End ----------------
|