Register limb masking buffer even if null

spaceml-org · Nov 15, 2024 · d86902e · d86902e
1 parent cb6c76b
commit d86902e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sdofm/models/mae3d.py b/sdofm/models/mae3d.py
@@ -93,8 +93,7 @@ def __init__(
 
         # --------------------------------------------------------------------------
         # Limb masking
-        if ids_limb_mask is not None:
-            self.register_buffer("ids_limb_mask", ids_limb_mask)
+        self.register_buffer("ids_limb_mask", ids_limb_mask)
 
         # MAE encoder specifics
         self.patch_embed = PatchEmbed(