Merge pull request #163 from facebookresearch/unify-loggers

[logging] Unify names for dinov3

Author: MichaelRamamonjisoa

dinov3/checkpointer/checkpointer.py

@@ -39,7 +39,7 @@
 import torch.distributed.checkpoint.state_dict as dcpsd
 from torch.distributed.checkpoint.stateful import Stateful
 
-logger = logging.getLogger(__name__)
+logger = logging.getLogger("dinov3")
 
 
 class CheckpointRetentionPolicy(Enum):

dinov3/configs/config.py

@@ -205,7 +205,7 @@ def setup_job(
         rank = distributed.get_rank()
         fix_random_seeds(seed + rank)
 
-    logger = logging.getLogger("fairvit")
+    logger = logging.getLogger("dinov3")
     logger.info("git:\n  {}\n".format(get_sha()))
 
     # Log some python info

dinov3/data/datasets/image_net.py

@@ -14,7 +14,7 @@
 from .decoders import ImageDataDecoder, TargetDecoder
 from .extended import ExtendedVisionDataset
 
-logger = logging.getLogger("dinov2")
+logger = logging.getLogger("dinov3")
 _Target = int
 
 

dinov3/eval/accumulators.py

@@ -12,7 +12,7 @@
 
 from dinov3.distributed import gather_all_tensors  # Gathers tensors of different sizes
 
-logger = logging.getLogger("fairvit")
+logger = logging.getLogger("dinov3")
 
 
 def _cat_and_gather_tensor_list(tensor_list: List[Tensor]) -> Tensor:

dinov3/eval/data.py

@@ -14,7 +14,7 @@
 
 from dinov3.eval.utils import extract_features
 
-logger = logging.getLogger("fairvit")
+logger = logging.getLogger("dinov3")
 
 
 class SubsetEx(Subset):

dinov3/eval/dense/depth/models/encoder.py

@@ -10,7 +10,7 @@
 from dinov3.eval.dense.depth.models.embed import CenterPadding, StretchToMultiple
 from torch import Tensor, nn
 
-logger = logging.getLogger("fairvit")
+logger = logging.getLogger("dinov3")
 
 
 class BackboneLayersSet(Enum):

dinov3/eval/log_regression.py

@@ -36,7 +36,7 @@
 from dinov3.run.init import job_context
 from dinov3.utils.dtype import as_torch_dtype
 
-logger = logging.getLogger("fairvit")
+logger = logging.getLogger("dinov3")
 
 
 RESULTS_FILENAME = "results-log-regression.csv"

dinov3/eval/metrics/classification.py

@@ -24,7 +24,7 @@
 
 from .imagenet_c import ImageNet_C_Metric
 
-logger = logging.getLogger("fairvit")
+logger = logging.getLogger("dinov3")
 
 
 class ClassificationMetricType(Enum):

dinov3/eval/metrics/imagenet_c.py

@@ -11,7 +11,7 @@
 from torch import Tensor
 from torchmetrics import Metric
 
-logger = logging.getLogger("fairvit")
+logger = logging.getLogger("dinov3")
 
 
 # corruption type (str) -> level (int) -> score (float)

dinov3/eval/results.py

@@ -14,7 +14,7 @@
 import pandas as pd
 import yaml  # type: ignore
 
-logger = logging.getLogger("fairvit")
+logger = logging.getLogger("dinov3")
 
 
 # This type represents a list of results, e.g. baselines for an evaluation.