Method__init__(
self,
dataset,
batch_size,
pool_size,
min_view_size,
max_vie
src/dust3r/datasets/base/batched_sampler.py:19
Method__init__(
self, net, has_conf=False, has_rgb=False, has_pose=False, mlp_ratio=4.0
)
src/dust3r/heads/linear_head.py:252
Method__init__(
self,
crop_size,
scale_prob=0.5,
scale_xonly=True,
lhth=800.0,
src/croco/stereoflow/augmentor.py:29
Method__init__(
self, dnames, trfs="", totensor=True, normalize=True, data_dir="./data/"
)
src/croco/datasets/pairs_dataset.py:128
Method__init__(
self,
scene,
navmesh,
scene_dataset_config_file,
resolution=(240, 32
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:145
Method__init__(
self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0.0, proj_drop=0.0
)
src/croco/models/blocks.py:98
Method__init__(
self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0.0, proj_drop=0.0
)
src/croco/models/blocks.py:195
Method__init__(
self,
*,
hooks_idx=None,
layer_dims=[96, 192, 384, 768],
output_widt
src/croco/models/head_downstream.py:26