From 36b98075eb297f2dd7588036c6c99c074582568d Mon Sep 17 00:00:00 2001 From: Joao Lucas de Sousa Almeida Date: Sat, 7 Dec 2024 12:41:31 -0300 Subject: [PATCH] info not debug Signed-off-by: Joao Lucas de Sousa Almeida --- terratorch/models/backbones/select_patch_embed_weights.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/terratorch/models/backbones/select_patch_embed_weights.py b/terratorch/models/backbones/select_patch_embed_weights.py index 5e9a2b77..142814d1 100644 --- a/terratorch/models/backbones/select_patch_embed_weights.py +++ b/terratorch/models/backbones/select_patch_embed_weights.py @@ -61,7 +61,7 @@ def select_patch_embed_weights( torch.nn.init.xavier_uniform_(temp_weight.view([temp_weight.shape[0], -1])) for index, band in enumerate(model_bands): if band in pretrained_bands: - logging.debug(f"Loaded weights for {band} in position {index} of patch embed") + logging.info(f"Loaded weights for {band} in position {index} of patch embed") temp_weight[:, index] = patch_embed_weight[:, pretrained_bands.index(band)] else: warnings.warn(