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