2025-03-07 19:20:37 +08:00
|
|
|
|
python train.py \
|
2025-03-18 21:12:32 +08:00
|
|
|
|
--dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Single/Monitor \
|
|
|
|
|
|
--name cp_1 \
|
|
|
|
|
|
--dataset_mode unaligned \
|
|
|
|
|
|
--display_env NEWCP \
|
|
|
|
|
|
--model roma_unsb_single \
|
|
|
|
|
|
--lambda_ctn 10 \
|
2025-03-15 15:00:00 +08:00
|
|
|
|
--lambda_inc 8.0 \
|
2025-03-07 19:20:37 +08:00
|
|
|
|
--lambda_global 6.0 \
|
|
|
|
|
|
--lambda_spatial 6.0 \
|
|
|
|
|
|
--gamma_stride 20 \
|
2025-03-18 21:12:32 +08:00
|
|
|
|
--lr 0.000002 \
|
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
|
2025-03-15 15:00:00 +08:00
|
|
|
|
# cp2 修了一下cp1的代码，--lr 0.000002
|
|
|
|
|
|
# cp3 加了--lambda_inc 8.0 --gpu_id 2
|
|
|
|
|
|
# cp4 在cp3的基础上把梯度增强给到了生成器中的ganloss --gpu_id 1
|
2025-03-18 21:12:32 +08:00
|
|
|
|
# cp5 在cp3的基础上，--lambda_ctn 0 ，--gpu_id 0.--lr 0.000005
|
|
|
|
|
|
# # newcp1 重新调整了光流算法，并且弄成单帧的脚本了，这一次是最终的复现了。--gpu_id 0
|
|
|
|
|
|
# # newcp2 把梯度图对loss的影响同样加到了G_GAN中。
|
