2025-02-23 15:57:25 +08:00
|
|
|
----------------- Options ---------------
|
2025-02-23 23:15:25 +08:00
|
|
|
atten_layers: 5
|
2025-02-23 15:57:25 +08:00
|
|
|
batch_size: 1
|
|
|
|
|
beta1: 0.5
|
|
|
|
|
beta2: 0.999
|
|
|
|
|
checkpoints_dir: ./checkpoints
|
|
|
|
|
continue_train: False
|
|
|
|
|
crop_size: 256
|
|
|
|
|
dataroot: /home/openxs/kunyu/datasets/InfraredCity-Lite/Double/Moitor [default: placeholder]
|
|
|
|
|
dataset_mode: unaligned_double [default: unaligned]
|
|
|
|
|
direction: AtoB
|
|
|
|
|
display_env: ROMA [default: main]
|
|
|
|
|
display_freq: 50
|
|
|
|
|
display_id: None
|
|
|
|
|
display_ncols: 4
|
|
|
|
|
display_port: 8097
|
|
|
|
|
display_server: http://localhost
|
|
|
|
|
display_winsize: 256
|
|
|
|
|
easy_label: experiment_name
|
|
|
|
|
epoch: latest
|
|
|
|
|
epoch_count: 1
|
|
|
|
|
eta_ratio: 0.1
|
|
|
|
|
evaluation_freq: 5000
|
|
|
|
|
flip_equivariance: False
|
|
|
|
|
gan_mode: lsgan
|
|
|
|
|
gpu_ids: 0
|
|
|
|
|
init_gain: 0.02
|
|
|
|
|
init_type: xavier
|
|
|
|
|
input_nc: 3
|
|
|
|
|
isTrain: True [default: None]
|
2025-02-23 23:15:25 +08:00
|
|
|
lambda_D_ViT: 1.0
|
2025-02-23 15:57:25 +08:00
|
|
|
lambda_GAN: 8.0 [default: 1.0]
|
|
|
|
|
lambda_NCE: 8.0 [default: 1.0]
|
|
|
|
|
lambda_SB: 0.1
|
|
|
|
|
lambda_ctn: 1.0
|
2025-02-23 23:15:25 +08:00
|
|
|
lambda_global: 1.0
|
2025-02-23 15:57:25 +08:00
|
|
|
lambda_inc: 1.0
|
|
|
|
|
lmda_1: 0.1
|
|
|
|
|
load_size: 286
|
|
|
|
|
lr: 1e-05 [default: 0.0002]
|
|
|
|
|
lr_decay_iters: 50
|
|
|
|
|
lr_policy: linear
|
|
|
|
|
max_dataset_size: inf
|
|
|
|
|
model: roma_unsb [default: cut]
|
|
|
|
|
n_epochs: 100
|
|
|
|
|
n_epochs_decay: 100
|
|
|
|
|
n_layers_D: 3
|
2025-02-23 18:42:21 +08:00
|
|
|
n_mlp: 3
|
2025-02-23 15:57:25 +08:00
|
|
|
name: ROMA_UNSB_001 [default: experiment_name]
|
|
|
|
|
nce_T: 0.07
|
|
|
|
|
nce_idt: False [default: True]
|
|
|
|
|
nce_includes_all_negatives_from_minibatch: False
|
|
|
|
|
nce_layers: 0,4,8,12,16
|
|
|
|
|
ndf: 64
|
2025-02-23 23:15:25 +08:00
|
|
|
netD: basic_cond
|
2025-02-23 15:57:25 +08:00
|
|
|
netF: mlp_sample
|
|
|
|
|
netF_nc: 256
|
2025-02-23 18:42:21 +08:00
|
|
|
netG: resnet_9blocks_cond
|
2025-02-23 15:57:25 +08:00
|
|
|
ngf: 64
|
|
|
|
|
no_antialias: False
|
|
|
|
|
no_antialias_up: False
|
|
|
|
|
no_dropout: True
|
|
|
|
|
no_flip: True [default: False]
|
|
|
|
|
no_html: False
|
|
|
|
|
normD: instance
|
|
|
|
|
normG: instance
|
|
|
|
|
num_patches: 256
|
|
|
|
|
num_threads: 4
|
2025-02-23 18:42:21 +08:00
|
|
|
num_timesteps: 10 [default: 5]
|
2025-02-23 15:57:25 +08:00
|
|
|
output_nc: 3
|
|
|
|
|
phase: train
|
|
|
|
|
pool_size: 0
|
|
|
|
|
preprocess: resize_and_crop
|
|
|
|
|
pretrained_name: None
|
|
|
|
|
print_freq: 100
|
|
|
|
|
random_scale_max: 3.0
|
|
|
|
|
save_by_iter: False
|
|
|
|
|
save_epoch_freq: 5
|
|
|
|
|
save_latest_freq: 5000
|
|
|
|
|
serial_batches: False
|
|
|
|
|
stylegan2_G_num_downsampling: 1
|
|
|
|
|
suffix:
|
2025-02-23 23:15:25 +08:00
|
|
|
tau: 0.01
|
2025-02-23 15:57:25 +08:00
|
|
|
update_html_freq: 1000
|
|
|
|
|
use_idt: False
|
|
|
|
|
verbose: False
|
|
|
|
|
----------------- End -------------------
|
