bishe 102595c040 File "/home/openxs/jj/roma_unsb/models/roma_unsb_model.py", line 421, in forward
bs = self.mutil_real_A0_tokens.size(0)
AttributeError: 'list' object has no attribute 'size'
2025-02-23 16:04:38 +08:00

85 lines
5.0 KiB
Plaintext

----------------- Options ---------------
atten_layers: 1,3,5
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]
lambda_GAN: 8.0 [default: 1.0]
lambda_NCE: 8.0 [default: 1.0]
lambda_SB: 0.1
lambda_ctn: 1.0
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
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
netD: basic
netF: mlp_sample
netF_nc: 256
netG: resnet_9blocks
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
num_timesteps: 10
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:
tau: 0.1
update_html_freq: 1000
use_idt: False
verbose: False
----------------- End -------------------