↓ 3 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, bias=True, drop=0.)
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:63
↓ 2 callersMethodprepare_tokens_with_masks(self, x_img, x_depth, x_img_mask=None, x_depth_mask=None, masks=None, **kwargs)
mdm/model/dinov2_rgbd/models/vision_transformer.py:276
↓ 1 callersFunction_build_mlp(nlayers, in_dim, bottleneck_dim, hidden_dim=None, use_bn=False, bias=True)
mdm/model/dinov2_rgbd/layers/dino_head.py:44
↓ 1 callersMethod_get_intermediate_layers_chunked(self, x_img, x_depth, x_img_mask=None, x_depth_mask=None, n=1, return_mae_aux=False, **kwargs)
mdm/model/dinov2_rgbd/models/vision_transformer.py:354
↓ 1 callersMethod_get_intermediate_layers_not_chunked(self, x_img, x_depth, x_img_mask=None, x_depth_mask=None, n=1, return_mae_aux=False, **kwargs)
mdm/model/dinov2_rgbd/models/vision_transformer.py:335
Method__init__(self, backbone: str, intermediate_layers: Union[int, List[int]], dim_out: int, ignore_layers: Union[str, List
mdm/model/modules_rgbd_encoder.py:23
Method__init__(self, base_value, final_value, total_iters, warmup_iters=0, start_warmup_value=0, freeze_iters=0)
mdm/model/dinov2_rgbd/utils/utils.py:68
Method__init__(
self,
img_size: Union[int, Tuple[int, int]] = 224,
patch_size: Union[int, Tuple[int,
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:96
Method__init__(
self,
in_dim,
out_dim,
use_bn=False,
nlayers=3,
hidden_dim=2
mdm/model/dinov2_rgbd/layers/dino_head.py:13
Method__init__(
self,
img_size: Union[int, Tuple[int, int]] = 224,
patch_size: Union[int, Tuple[int,
mdm/model/dinov2_rgbd/layers/patch_embed.py:37