info not debug

Signed-off-by: Joao Lucas de Sousa Almeida <[email protected]>
IBM · Dec 7, 2024 · 36b9807 · 36b9807
1 parent fc1bc25
commit 36b9807
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(