-
Ponku authored
* crestereo draft implementation * minor model fixes. positional embedding changes. * aligned base configuration with paper * Adressing comments * Broke down Adaptive Correlation Layer. Adressed some other commets. * adressed some nits * changed search size, added output channels to model attrs * changed weights naming * changed from iterations to num_iters * removed _make_coords, adressed comments * fixed jit test * added script files * added cascaded inference evaluation * added optimizer option * minor changes * Update references/depth/stereo/train.py Co-authored-by:
vfdev <vfdev.5@gmail.com> * adressed some comments * change if-else to dict * added manual resizing for masks and disparities during evaluation * minor fixes after previous changes * changed dataloader to be initialised once * added distributed changes * changed loader logic * updated eval script to generate weight API like logs * improved support for fine-tuning / training resume * minor changes for finetuning * updated with transforms from main * logging distributed deadlock fix * lint fix * updated metrics * weights API log support * lint fix * added readme * updated readme * updated readme * read-me update * remove hardcoded paths. improved valid dataset selection and sync * removed extras from gitignore Co-authored-by:
Joao Gomes <jdsgomes@fb.com> Co-authored-by:
vfdev <vfdev.5@gmail.com> Co-authored-by:
YosuaMichael <yosuamichaelm@gmail.com>
10dafd9b