roma_unsb/scripts/traincp.sh

21 lines
709 B
Bash
Raw Normal View History

2025-03-07 19:20:37 +08:00
python train.py \
--dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Double/Moitor \
--name cp_5 \
2025-03-07 19:20:37 +08:00
--dataset_mode unaligned_double \
--display_env CP \
--model roma_unsb \
--lambda_ctn 0 \
--lambda_inc 8.0 \
2025-03-07 19:20:37 +08:00
--lambda_global 6.0 \
--lambda_spatial 6.0 \
--gamma_stride 20 \
--lr 0.000005 \
2025-03-09 21:41:52 +08:00
--gpu_id 0 \
2025-03-07 19:20:37 +08:00
--eta_ratio 0.4 \
--n_epochs 100 \
--n_epochs_decay 100 \
2025-03-09 21:41:52 +08:00
# cp1 复现cptrans的效果 --lr 0.000001
# cp2 修了一下cp1的代码--lr 0.000002
# cp3 加了--lambda_inc 8.0 --gpu_id 2
# cp4 在cp3的基础上把梯度增强给到了生成器中的ganloss --gpu_id 1
# cp5 在cp3的基础上--lambda_ctn 0 --gpu_id 0.--lr 0.000005