20 lines
645 B
Bash
20 lines
645 B
Bash
python train.py \
|
||
--dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Double/Moitor \
|
||
--name cp_3 \
|
||
--dataset_mode unaligned_double \
|
||
--display_env CP \
|
||
--model roma_unsb \
|
||
--lambda_ctn 10 \
|
||
--lambda_inc 8.0 \
|
||
--eta_ratio 0.4 \
|
||
--lambda_global 6.0 \
|
||
--lambda_spatial 6.0 \
|
||
--gamma_stride 20 \
|
||
--lr 0.00002 \
|
||
--gpu_id 3 \
|
||
--eta_ratio 0.4 \
|
||
--n_epochs 100 \
|
||
--n_epochs_decay 100 \
|
||
# cp1 复现cptrans的效果 --lr 0.000001
|
||
# cp2 修了一下cp1的代码，--lr 0.000002
|
||
## cp3 将梯度加强修改为attention加强，--lr 0.000005,--lambda_inc 8.0,--gpu_id 3(基于cp2的sh) |