55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
"""Gin-configurable settings for the caption-test project.
|
|
|
|
Five universal axes of variability:
|
|
- PipelineConfig — paths, training schedule, output, resume
|
|
- HardwareConfig — batch size, accumulation, AMP, gradient checkpointing
|
|
- TrainingConfig — loss + optimizer + sampler (the recipe)
|
|
- TrackingConfig — wandb / tensorboard / gradcam / profiler
|
|
|
|
Plus a model-family config: ModelsCommonConfig describes the active backbone,
|
|
and one of {DINOv3, StripNet, SOFIAv1, SOFIAv71} ModelsConfig classes
|
|
parameterises it.
|
|
|
|
Plus PreprocessConfig used only by scripts/make_split.py and
|
|
scripts/filter_segmentation.py.
|
|
|
|
All configs are loaded together via load_all_configs(path2cfg) — see
|
|
config_loader.py.
|
|
"""
|
|
|
|
from src.conf.config_loader import load_all_configs
|
|
from src.conf.hardware_conf import HardwareConfig, get_hardware_cfg
|
|
from src.conf.models_common_conf import ModelsCommonConfig, get_models_common_cfg
|
|
from src.conf.models_dinov3_conf import DINOv3ModelsConfig, get_models_dinov3_cfg
|
|
from src.conf.models_sofia_v1_conf import SOFIAv1ModelsConfig, get_models_sofia_v1_cfg
|
|
from src.conf.models_sofia_v71_conf import SOFIAv71ModelsConfig, get_models_sofia_v71_cfg
|
|
from src.conf.models_stripnet_conf import StripNetModelsConfig, get_models_stripnet_cfg
|
|
from src.conf.pipeline_conf import PipelineConfig, get_pipeline_cfg
|
|
from src.conf.preprocess_conf import PreprocessConfig, get_preprocess_cfg
|
|
from src.conf.tracking_conf import TrackingConfig, get_tracking_cfg
|
|
from src.conf.training_conf import TrainingConfig, get_training_cfg
|
|
|
|
__all__ = [
|
|
"DINOv3ModelsConfig",
|
|
"HardwareConfig",
|
|
"ModelsCommonConfig",
|
|
"PipelineConfig",
|
|
"PreprocessConfig",
|
|
"SOFIAv1ModelsConfig",
|
|
"SOFIAv71ModelsConfig",
|
|
"StripNetModelsConfig",
|
|
"TrackingConfig",
|
|
"TrainingConfig",
|
|
"get_hardware_cfg",
|
|
"get_models_common_cfg",
|
|
"get_models_dinov3_cfg",
|
|
"get_models_sofia_v1_cfg",
|
|
"get_models_sofia_v71_cfg",
|
|
"get_models_stripnet_cfg",
|
|
"get_pipeline_cfg",
|
|
"get_preprocess_cfg",
|
|
"get_tracking_cfg",
|
|
"get_training_cfg",
|
|
"load_all_configs",
|
|
]
|