opt.txt

train.py 2023-07-08 16:16:10
------------------------------ Options -------------------------------
annotation_file_path: dataset/annotation.csv
batch_accum_size: 10
batch_size: 2
beta1: 0.5
checkpoint_dir: checkpoints
checkpoint_step: 500
contextual_loss_direction: forward
data_root_imagenet: imagenet
data_root_imagenet_full: imagenet
data_root_list: dataset
domain_invariant: False
epoch: 40
gpu_ids: 0,1,2,3
head_out_idx: 8,9,10,11
ic: 7
image_size: [384, 384]
imagenet_pairs_file: imagenet_pairs.txt
load_pretrained_model: False
lr: 0.0001
lr_gamma: 0.9
lr_step: 2000
luminance_noise: 2.0
nonzero_placeholder_probability: 0.0
permute_data: True
real_reference_probability: 0.7
resume: False
resume_epoch: 0
step_train_discriminator: 1
use_dummy: True
use_feature_transform: False
use_wandb: False
vit_version: vit_tiny_patch16_384
wandb_name: 
wandb_token: 
weight_consistent: 0.05
weight_contextual: 0.5
weight_gan: 0.5
weight_nonlocal_consistent: 0.0
weight_nonlocal_smoothness: 0.0
weight_perceptual: 0.02
weight_smoothness: 5.0
weigth_l1: 2.0
workers: 0
-------------------------------- End ---------------------------------