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(