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

remove redundant: train.py: error: unrecognized arguments: --local_nums 64 --side_length 2 --atten_l

Browse Source
This commit is contained in:
123456 2025-02-23 15:42:21 +08:00
parent 6317aeaa11
commit 4c6ad27e16
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/train.sh
View File

@ -10,16 +10,13 @@ python train.py \
--name ROMA_UNSB_001 \ --name ROMA_UNSB_001 \
--dataset_mode unaligned_double \ --dataset_mode unaligned_double \
--no_flip \ --no_flip \
--local_nums 64 \
--display_env ROMA \ --display_env ROMA \
--model roma_unsb \ --model roma_unsb \
--side_length 2 \
--lambda_GAN 8.0 \ --lambda_GAN 8.0 \
--lambda_NCE 8.0 \ --lambda_NCE 8.0 \
--lambda_SB 0.1 \ --lambda_SB 0.1 \
--lambda_ctn 1.0 \ --lambda_ctn 1.0 \
--lambda_inc 1.0 \ --lambda_inc 1.0 \
--atten_layers 5 \
--lr 0.00001 \ --lr 0.00001 \
--gpu_id 0 \ --gpu_id 0 \
--nce_idt False \ --nce_idt False \