mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-12 12:25:32 +08:00
feat(utils): add feature extraction utilities and tests
This commit is contained in:
@@ -30,6 +30,6 @@ benchmark:
|
||||
task:
|
||||
name: "recall_at_k"
|
||||
type: "retrieval"
|
||||
top_k: 10
|
||||
top_k: 1
|
||||
batch_size: 64
|
||||
model_table_prefix: "benchmark"
|
||||
|
||||
@@ -11,7 +11,7 @@ class ModelConfig(BaseModel):
|
||||
|
||||
model_config = ConfigDict(extra="ignore")
|
||||
|
||||
name: str = "facebook/dinov2-large"
|
||||
dino_model: str = "facebook/dinov2-large"
|
||||
compression_dim: int = Field(
|
||||
default=512, gt=0, description="Output feature dimension"
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user