1. 15 Sep, 2022 1 commit
    • 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