123456/roma_unsb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cptrans复现完成

Browse Source
This commit is contained in:
bishe 2025-03-22 15:27:19 +08:00
parent c173e29ea6
commit 77468f16f9
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/test.sh
View File

@ -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
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

4
scripts/traincp.sh
View File

@ -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 \