diff --git a/scripts/test.sh b/scripts/test.sh
index 7f91de0..46b0108 100755
--- a/scripts/test.sh
+++ b/scripts/test.sh
@@ -1,2 +1,2 @@
 # CUDA_VISIBLE_DEVICES=0 python test.py --dataroot /path/of/test_dataset --checkpoints_dir ./checkpoints --name train1 --model roma_single --num_test 10000 --epoch latest 
-python test.py --dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Single/Monitor --checkpoints_dir ./checkpoints --name cp_4 --model roma_single --num_test 4132 --epoch 150  --gpu_id 2
\ No newline at end of file
+python test.py --dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Single/Monitor --checkpoints_dir ./checkpoints --name cp_2 --model roma_single --num_test 4132 --epoch 120  --gpu_id 1
\ No newline at end of file
diff --git a/scripts/traincp.sh b/scripts/traincp.sh
index 6ebb804..24e7684 100644
--- a/scripts/traincp.sh
+++ b/scripts/traincp.sh
@@ -1,6 +1,6 @@
 python train.py \
     --dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Single/Monitor \
-    --name cp_1 \
+    --name cp_2 \
     --dataset_mode unaligned \
     --display_env NEWCP \
     --model roma_unsb_single \
@@ -10,7 +10,7 @@ python train.py \
     --lambda_spatial 6.0 \
     --gamma_stride 20 \
     --lr 0.000002 \
-    --gpu_id 0 \
+    --gpu_id 1 \
     --eta_ratio 0.4 \
     --n_epochs 100 \
     --n_epochs_decay 100 \