1. 15 Mar, 2022 1 commit
    • YosuaMichael's avatar
      RAFT training reference Improvement (#5590) · 3aa2a93d
      YosuaMichael authored
      
      
      * Change optical flow train.py function name from validate to evaluate so it is similar to other references
      
      * Add --device as parameter and enable to run in non distributed mode
      
      * Format with ufmt
      
      * Fix unneccessary param and bug
      
      * Enable saving the optimizer and scheduler on the checkpoint
      
      * Fix bug when evaluate before resume and save or load model without ddp
      
      * Fix case where --train-dataset is None
      Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      3aa2a93d
  2. 07 Mar, 2022 1 commit
  3. 24 Feb, 2022 1 commit
  4. 21 Jan, 2022 1 commit
  5. 13 Dec, 2021 1 commit
  6. 09 Dec, 2021 2 commits
  7. 08 Dec, 2021 1 commit
  8. 07 Dec, 2021 2 commits