Method__init__(self, crop_size, scale_prob=0.5, scale_xonly=True, lhth=800., lminscale=0.0, lmaxscale=1.0, hminscale=-0.2, h
Dens3R/croco/stereoflow/augmentor.py:27
Method__init__(self, crop_size, min_scale=-0.2, max_scale=0.5, spatial_aug_prob=0.8, stretch_prob=0.8, max_stretch=0.2, h_fl
Dens3R/croco/stereoflow/augmentor.py:163
Method__init__(self, dnames, trfs='', totensor=True, normalize=True, data_dir='./data/')
Dens3R/croco/datasets/pairs_dataset.py:87
Method__init__(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)
Dens3R/croco/models/blocks.py:83
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=0., attn_drop=0.,
drop_path=0., act
Dens3R/croco/models/blocks.py:116
Method__init__(self, dim, num_heads, qkv_bias=False, attn_drop=0., drop=0., act_layer=nn.GELU, norm_layer=nn.LayerNorm, mlp_
Dens3R/croco/models/blocks.py:134
Method__init__(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)
Dens3R/croco/models/blocks.py:169
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=0., attn_drop=0.,
drop_path=0., act
Dens3R/croco/models/blocks.py:208
Method__init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True)
Dens3R/croco/models/blocks.py:247
Method__init__(self, *, hooks_idx=None, layer_dims=[96,192,384,768],
output_width_ratio=1, num_channels=1,
Dens3R/croco/models/head_downstream.py:26