From 77468f16f99a97681d4fd0a24a1c7dbc5bded513 Mon Sep 17 00:00:00 2001
From: bishe <123456789@163.com>
Date: Sat, 22 Mar 2025 15:27:19 +0800
Subject: [PATCH] =?UTF-8?q?cptrans=E5=A4=8D=E7=8E=B0=E5=AE=8C=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/test.sh    | 2 +-
 scripts/traincp.sh | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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 \