From 55335dff1666b1013b8e743fa0083bcf2bd1e60f Mon Sep 17 00:00:00 2001
From: 123456 <3351416005@qq.com>
Date: Sun, 23 Feb 2025 15:46:18 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20models/roma=5Funsb=5Fmodel?=
 =?UTF-8?q?.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/roma_unsb_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py
index 046cbdd..b4df892 100644
--- a/models/roma_unsb_model.py
+++ b/models/roma_unsb_model.py
@@ -218,6 +218,8 @@ class RomaUnsbModel(BaseModel):
         
         parser.add_argument('--lambda_inc', type=float, default=1.0, help='incremental weight for content-aware optimization')
         parser.add_argument('--eta_ratio', type=float, default=0.1, help='ratio of content-rich regions')
+
+        parser.add_argument('--atten_layers', type=str, default='1,3,5', help='compute Cross-Similarity on which layers')
         
 
         parser.set_defaults(pool_size=0)  # no image pooling