17 lines
455 B
Bash
17 lines
455 B
Bash
python train.py \
|
|
--dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Double/Moitor \
|
|
--name cp_1 \
|
|
--dataset_mode unaligned_double \
|
|
--display_env CP \
|
|
--model roma_unsb \
|
|
--lambda_ctn 10 \
|
|
--lambda_inc 1.0 \
|
|
--lambda_global 6.0 \
|
|
--lambda_spatial 6.0 \
|
|
--gamma_stride 20 \
|
|
--lr 0.000001 \
|
|
--gpu_id 2 \
|
|
--eta_ratio 0.4 \
|
|
--n_epochs 100 \
|
|
--n_epochs_decay 100 \
|
|
# cp1 复现cptrans的效果 |