From db2b5b32f44f0dadd64d21b55706c4596b483752 Mon Sep 17 00:00:00 2001 From: pikaliov Date: Wed, 29 Apr 2026 16:38:20 +0300 Subject: [PATCH] checkpoint 29_04 --- src/conf/models_sofia_v1_conf.py | 70 ++++++++++++++++---------------- 1 file changed, 35 insertions(+), 35 deletions(-) diff --git a/src/conf/models_sofia_v1_conf.py b/src/conf/models_sofia_v1_conf.py index 525b5c4..658ae79 100644 --- a/src/conf/models_sofia_v1_conf.py +++ b/src/conf/models_sofia_v1_conf.py @@ -1,37 +1,37 @@ -import gin +# 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 -@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 -