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

更新 models/roma_unsb_model.py

Browse Source
This commit is contained in:
123456 2025-02-23 15:46:18 +08:00
parent 4c6ad27e16
commit 55335dff16
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/roma_unsb_model.py
View File

@ -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('--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('--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 parser.set_defaults(pool_size=0) # no image pooling