Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer, model)
stereo/modeling/trainer_template.py:22
Method__init__(self, init_fmap1, init_fmap2, geo_volume, num_levels=2, radius=4)
stereo/modeling/disp_refinement/gru_blocks.py:171
Method__init__(self, n_gru_layers, corr_levels, corr_radius, volume_channel, hidden_dims=None)
stereo/modeling/disp_refinement/gru_blocks.py:291
Method__init__(self, in_channels, out_channels, kernel_size, stride=1, padding=0, bias=False,
norm_layer=No
stereo/modeling/common/basic_block_3d.py:24
Method__init__(self, in_channels, out_channels, kernel_size, stride=1, padding=0, bias=False,
norm_layer=No
stereo/modeling/common/basic_block_2d.py:25
Method__init__(self, inp, oup, stride, expanse_ratio, dilation=1)
stereo/modeling/models/lightstereo/aggregation.py:66
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/lightstereo/trainer.py:12
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/sttr/trainer.py:11
Method__init__(self, left, right, disp=None, sampled_cols=None, sampled_rows=None, occ_mask=None,
occ_mask_
stereo/modeling/models/sttr/utilities/misc.py:13
Method__init__(self, num_layers, num_input_features, bn_size, growth_rate, drop_rate, memory_efficient=False)
stereo/modeling/models/sttr/utilities/densenet_in.py:97
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, **kwargs)
stereo/modeling/models/igev_rt/submodule.py:11
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, IN=True, relu=True, **kwargs)
stereo/modeling/models/igev_rt/submodule.py:82
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, IN=True, relu=True
stereo/modeling/models/igev_rt/submodule.py:111
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/igev_rt/trainer.py:16
Method__init__(self, init_fmap1, init_fmap2, geo_volume, num_levels=2, radius=4)
stereo/modeling/models/stereobase/gru_blocks.py:171
Method__init__(self, n_gru_layers, corr_levels, corr_radius, volume_channel, hidden_dims=None)
stereo/modeling/models/stereobase/gru_blocks.py:291
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/stereobase/trainer.py:12
Method__init__(self, max_disparity, in_chan, hidden_chan, head=1, weighted=False)
stereo/modeling/models/coex/coex_cost_processor.py:39
Method__init__(self, inplanes, planes, stride, downsample, pad, dilation)
stereo/modeling/models/coex/submodule.py:20
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, **kwargs)
stereo/modeling/models/coex/submodule.py:45
Method__init__(self, in_channels, out_channels, stride=1, bn=True, relu=True)
stereo/modeling/models/coex/submodule.py:278