diff --git a/scripts/train.sh b/scripts/train.sh
index e4643bc..3315465 100755
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -10,16 +10,13 @@ python train.py \
     --name ROMA_UNSB_001 \
     --dataset_mode unaligned_double \
     --no_flip \
-    --local_nums 64 \
     --display_env ROMA \
     --model roma_unsb \
-    --side_length 2 \
     --lambda_GAN 8.0 \
     --lambda_NCE 8.0 \
     --lambda_SB 0.1 \
     --lambda_ctn 1.0 \
     --lambda_inc 1.0 \
-    --atten_layers 5 \
     --lr 0.00001 \
     --gpu_id 0 \
     --nce_idt False \