更新 models/roma_unsb_model.py

2025-02-23 15:46:18 +08:00 · 2025-02-23 15:46:18 +08:00 · 55335dff16
commit 55335dff16
parent 4c6ad27e16
1 changed files with 2 additions and 0 deletions
--- a/models/roma_unsb_model.py
+++ b/models/roma_unsb_model.py
@ -219,6 +219,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