1. 23 Sep, 2022 1 commit
    • 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
  2. 07 Jul, 2022 1 commit
    • YosuaMichael's avatar
      Adding video accuracy for video_classification reference script (#6241) · 8a45147f
      YosuaMichael authored
      * Add ensembled video accuracy on video reference script
      
      * Change the parser func to be similar with classification reference
      
      * Fix typo type->dtype
      
      * Use custom kinetics
      
      * Fix dataset to not getting start_pts
      
      * Change dataset name, and put video_idx at the back
      
      * Ufmt format
      
      * Use functional softmax, updating meta and use it to overwrite eval param
      
      * Fix typo
      
      * Put the eval parameters on the docs for now
      
      * Change meta for video resnet to use frame-rate 15, also change wording on docs
      8a45147f
  3. 03 May, 2022 1 commit
  4. 15 Mar, 2022 1 commit
  5. 24 Feb, 2022 1 commit
  6. 28 Oct, 2021 1 commit
  7. 17 Oct, 2021 1 commit
  8. 04 Oct, 2021 1 commit
    • Philip Meier's avatar
      Add ufmt (usort + black) as code formatter (#4384) · 5f0edb97
      Philip Meier authored
      
      
      * add ufmt as code formatter
      
      * cleanup
      
      * quote ufmt requirement
      
      * split imports into more groups
      
      * regenerate circleci config
      
      * fix CI
      
      * clarify local testing utils section
      
      * use ufmt pre-commit hook
      
      * split relative imports into local category
      
      * Revert "split relative imports into local category"
      
      This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51.
      
      * pin black and usort dependencies
      
      * fix local test utils detection
      
      * fix ufmt rev
      
      * add reference utils to local category
      
      * fix usort config
      
      * remove custom categories sorting
      
      * Run pre-commit without fixing flake8
      
      * got a double import in merge
      Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      5f0edb97
  9. 31 Mar, 2020 1 commit
  10. 31 Jul, 2019 1 commit
    • Francisco Massa's avatar
      Video reference scripts (#1180) · 5c0b7f31
      Francisco Massa authored
      * Copy classification scripts for video classification
      
      * Initial version of video classification
      
      * add version
      
      * Training of r2plus1d_18 on kinetics work
      
      Gives even slightly better results than expected, with 57.336 top1 clip accuracy. But we count some clips twice in this evaluation
      
      * Cleanups on training script
      
      * Lint
      
      * Minor improvements
      
      * Remove some hacks
      
      * Lint
      5c0b7f31
  11. 14 Jun, 2019 1 commit
    • LXYTSOS's avatar
      utils.py in references can't work with pytorch-cpu (#1023) · 250bac89
      LXYTSOS authored
      * can't work with pytorch-cpu fixed
      
      utils.py can't work with pytorch-cpu because of this line of code `memory=torch.cuda.max_memory_allocated()`
      
      * can't work with pytorch-cpu fixed
      
      utils.py can't work with pytorch-cpu because of this line of code 'memory=torch.cuda.max_memory_allocated()'
      250bac89
  12. 08 May, 2019 1 commit
  13. 02 Apr, 2019 2 commits
  14. 28 Mar, 2019 1 commit
    • Francisco Massa's avatar
      Initial version of classification reference scripts (#819) · 27ff89f6
      Francisco Massa authored
      * Initial version of classification reference training script
      
      * Updates
      
      * Minor updates
      
      * Expose a few more options
      
      * Load optimizer and lr_scheduler when resuming
      
      Also log the learning rate
      
      * Evaluation-only and minor improvements
      
      Identified a bug in the reporting of the results. They need to be reduced between all processes
      
      * Address Soumith's comment
      
      * Fix some approximations on the evaluation metric
      
      * Flake8
      27ff89f6