• Ponku's avatar
    add crestereo implementation (#6310) · 1c3eedc4
    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
    
    * config nit
    
    * Changed device arg to str
    Co-authored-by: default avatarJoao Gomes <jdsgomes@fb.com>
    Co-authored-by: default avatarYosuaMichael <yosuamichaelm@gmail.com>
    1c3eedc4
_utils.py 2.03 KB