• Ponku's avatar
    Add stereo train loop (#6605) · 10dafd9b
    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: default avatarvfdev <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: default avatarJoao Gomes <jdsgomes@fb.com>
    Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
    Co-authored-by: default avatarYosuaMichael <yosuamichaelm@gmail.com>
    10dafd9b
__init__.py 136 Bytes