claude_refactor_v3: Added models configs except sofiav71 - need extra presets for M, L, Tiny
This commit is contained in:
@@ -1 +1,18 @@
|
|||||||
|
import gin
|
||||||
|
|
||||||
|
@gin.configurable
|
||||||
|
class ModelsCommonConfig:
|
||||||
|
"""Common architecture switches shared by all backbones."""
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
backbone: str = "dinov3", # 'dinov3' | 'stripnet' | 'sofia_v1' | 'sofia_v71'
|
||||||
|
baseline_mode: bool = False, # text disabled, gate forced 1.0
|
||||||
|
init_gate: float = 0.7,
|
||||||
|
lrsclip_path: str = "nn_models/LRSCLIP/DGTRS-CLIP-ViT-L-14.pt",
|
||||||
|
) -> None:
|
||||||
|
self.backbone = backbone
|
||||||
|
self.baseline_mode = baseline_mode
|
||||||
|
self.init_gate = init_gate
|
||||||
|
self.lrsclip_path = lrsclip_path
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1 +1,19 @@
|
|||||||
|
import gin
|
||||||
|
|
||||||
|
@gin.configurable
|
||||||
|
class DINOv3ModelsConfig:
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
dino_web_path: str = "nn_models/DINO_WEB/dinov3-vitl16-pretrain-lvd1689m.pth",
|
||||||
|
dino_sat_path: str = "nn_models/DINO_SAT/model.safetensors",
|
||||||
|
shared_encoder: bool = True,
|
||||||
|
mona_bottleneck: int = 64,
|
||||||
|
mona_last_n_blocks: int = 12,
|
||||||
|
) -> None:
|
||||||
|
self.dino_web_path = dino_web_path
|
||||||
|
self.dino_sat_path = dino_sat_path
|
||||||
|
self.shared_encoder = shared_encoder
|
||||||
|
self.mona_bottleneck = mona_bottleneck
|
||||||
|
self.mona_last_n_blocks = mona_last_n_blocks
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1 +1,37 @@
|
|||||||
|
import gin
|
||||||
|
|
||||||
|
@gin.configurable
|
||||||
|
class SOFIAv1ModelsConfig:
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
# Backbone.
|
||||||
|
variant: str = "small", # 'tiny_tiny' | 'tiny' | 'small' | 'small_v2'
|
||||||
|
in_channels: int = 3,
|
||||||
|
input_size: int = 256,
|
||||||
|
dcn_variant: str = "v2", # 'v2' (stable) | 'v4' (faster, leaks)
|
||||||
|
# Heads.
|
||||||
|
d_descriptor: int = 1024,
|
||||||
|
return_normalized: bool = False,
|
||||||
|
# Altitude-FiLM.
|
||||||
|
use_film_altitude: bool = True,
|
||||||
|
altitude_norm: float = 500.0,
|
||||||
|
# Text-FiLM.
|
||||||
|
use_text_film_uav: bool = True,
|
||||||
|
use_text_film_sat: bool = True,
|
||||||
|
text_film_dim: int = 1024,
|
||||||
|
text_film_hidden: int = 256,
|
||||||
|
) -> None:
|
||||||
|
self.variant = variant
|
||||||
|
self.in_channels = in_channels
|
||||||
|
self.input_size = input_size
|
||||||
|
self.dcn_variant = dcn_variant
|
||||||
|
self.d_descriptor = d_descriptor
|
||||||
|
self.return_normalized = return_normalized
|
||||||
|
self.use_film_altitude = use_film_altitude
|
||||||
|
self.altitude_norm = altitude_norm
|
||||||
|
self.use_text_film_uav = use_text_film_uav
|
||||||
|
self.use_text_film_sat = use_text_film_sat
|
||||||
|
self.text_film_dim = text_film_dim
|
||||||
|
self.text_film_hidden = text_film_hidden
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1 +0,0 @@
|
|||||||
|
|
||||||
|
|||||||
@@ -1 +1,16 @@
|
|||||||
|
import gin
|
||||||
|
|
||||||
|
@gin.configurable
|
||||||
|
class StripNetModelsConfig:
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
stripnet_path: str = "nn_models/STRIPNET/stripnet_s.pth",
|
||||||
|
stripnet_freeze: bool = True,
|
||||||
|
stripnet_mona_last_n_stages: int = 2,
|
||||||
|
stripnet_backbone_lr_factor: float = 0.1,
|
||||||
|
) -> None:
|
||||||
|
self.stripnet_path = stripnet_path
|
||||||
|
self.stripnet_freeze = stripnet_freeze
|
||||||
|
self.stripnet_mona_last_n_stages = stripnet_mona_last_n_stages
|
||||||
|
self.stripnet_backbone_lr_factor = stripnet_backbone_lr_factor
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user