Method__init__(self, channels, use_conv, dims=2, out_channels=None,padding=1,
downsample_3ddim=False, # downsample all
src/model/encoder/costvolume/ldm_unet/unet.py:149
Method__init__(
self,
channels,
emb_channels,
dropout,
out_channels=None,
us
src/model/encoder/costvolume/ldm_unet/unet.py:192
Method__init__(
self,
channels,
condition_channels,
num_heads=8,
proj_channels=512,
src/model/encoder/costvolume/ldm_unet/unet.py:391
Method__init__(
self,
image_size,
in_channels,
model_channels,
out_channels,
src/model/encoder/costvolume/ldm_unet/unet.py:637
Method__init__(self, dim, dim_out=None, mult=4, glu=False, dropout=0.)
src/model/encoder/costvolume/ldm_unet/attention.py:46
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.)
src/model/encoder/costvolume/ldm_unet/attention.py:151
Method__init__(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, checkpoint=False)
src/model/encoder/costvolume/ldm_unet/attention.py:195
Method__init__(self, in_channels, n_heads, d_head,
depth=1, dropout=0., context_dim=None)
src/model/encoder/costvolume/ldm_unet/attention.py:226
Method__init__(
self,
in_chs,
out_chs,
kernel_size=7,
stride=2,
depth=2,
src/model/unidepth/backbones/convnext.py:237
Method__init__(
self,
in_chans=3,
depths=[3, 3, 9, 3],
dims=96,
drop_path_rate=0.0,
src/model/unidepth/backbones/convnext2.py:206
Method__init__(
self,
in_dim,
out_dim,
use_bn=False,
nlayers=3,
hidden_dim=2
src/model/unidepth/backbones/metadinov2/dino_head.py:14
Method__init__(
self,
img_size: Union[int, Tuple[int, int]] = 224,
patch_size: Union[int, Tuple[int,
src/model/unidepth/backbones/metadinov2/patch_embed.py:38
Method__init__(
self,
dim,
pt_seq_len,
ft_seq_len=None,
custom_freqs=None,
f
src/model/unidepth/utils/positional_embedding.py:115
Method__init__(
self,
dim,
pt_seq_len,
ft_seq_len=None,
custom_freqs=None,
f
src/model/unidepth/utils/positional_embedding.py:173
Method__init__(
self,
dim,
kernel_size=7,
layer_scale=1.0,
expansion=4,
dila
src/model/unidepth/layers/convnext.py:6
Method__init__(
self,
dim,
pt_seq_len,
ft_seq_len=None,
custom_freqs=None,
f
src/model/unidepth/layers/positional_encoding.py:131
Method__init__(
self,
dim,
pt_seq_len,
ft_seq_len=None,
custom_freqs=None,
f
src/model/unidepth/layers/positional_encoding.py:189
Method__init__(
self,
in_channels,
out_channels,
kernel_size=3,
stride=1,
pa
src/model/LightGlue/lightglue/aliked.py:292